From 2f00d6ab37724f6f5a0466b20832b756812d8449 Mon Sep 17 00:00:00 2001
From: perkyfever <dteakhperky@gmail.com>
Date: Mon, 6 Oct 2025 01:18:35 +0000
Subject: [PATCH] Fix docstring & Add missing safetensors requirement

---
 requirements.txt        | 1 +
 src/utils/init_utils.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index f258620..7e9d118 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,6 +8,7 @@ pandas
 wandb
 comet_ml
 hydra-core
+safetensors
 
 black
 isort
diff --git a/src/utils/init_utils.py b/src/utils/init_utils.py
index 8c5aff7..aae2cec 100644
--- a/src/utils/init_utils.py
+++ b/src/utils/init_utils.py
@@ -142,7 +142,7 @@ def setup_saving_and_logging(config):
     """
     Initialize the logger, writer, and saving directory.
     The saving directory is defined by the run_name and save_dir
-    arguments of config.writer and config.trainer, respectfully.
+    arguments of config.writer and config.trainer, respectively.
 
     Args:
         config (DictConfig): hydra config for the current experiment.