old structure

dushyantbehl · Feb 19, 2024 · bf8ea61 · bf8ea61
1 parent 81f5087
commit bf8ea61
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/tuning/tracker/aim_loader.py → tuning/aim_loader.py b/tuning/tracker/aim_loader.py → tuning/aim_loader.py
@@ -21,7 +21,7 @@ def __init__(
         # Initialise the run_hash
         self._run_hash = run_hash
 
-def get_custom_aimstack_callback():
+def get_aimstack_callback():
     # Initialize a new run
     aim_server = os.environ.get("AIMSTACK_SERVER")
     aim_repo = os.environ.get("AIMSTACK_REPO")

diff --git a/tuning/sft_trainer.py b/tuning/sft_trainer.py
@@ -22,7 +22,7 @@
 import transformers
 
 # Local
-from tuning.tracker.aim_loader import get_custom_aimstack_callback
+from tuning.aim_loader import get_aimstack_callback
 from tuning.config import configs, peft_config
 from tuning.data import tokenizer_data_utils
 from tuning.utils.config_utils import get_hf_peft_config

diff --git a/tuning/tracker/__init__.py b/tuning/tracker/__init__.py