Bump aim version.

Fix duplicate argument Signed-off-by: Dushyant Behl <[email protected]>
dushyantbehl · Feb 20, 2024 · 2d5ffa5 · 2d5ffa5
1 parent e2a2e98
commit 2d5ffa5
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 9 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -2,7 +2,7 @@ numpy
 accelerate>=0.20.3
 transformers>=4.34.1
 torch
-aim==3.17.5
+aim==3.18.1
 sentencepiece
 tokenizers>=0.13.3
 tqdm

diff --git a/tuning/config/configs.py b/tuning/config/configs.py
@@ -57,7 +57,7 @@ class TrainingArguments(transformers.TrainingArguments):
         default=False,
         metadata={"help": "Packing to be enabled in SFT Trainer, default is False"},
     )
-    tracker: str = field(
+    tracker: str.lower = field(
         default="aim",
-        metadata={"help", "Default experiment tracker to integrate with. requires additional configs. see tuning.configs/tracker_configs.py"}
+        metadata={"help": "Default experiment tracker to integrate with. requires additional configs. see tuning.configs/tracker_configs.py"}
     )
diff --git a/tuning/sft_trainer.py b/tuning/sft_trainer.py
@@ -299,12 +299,6 @@ def main(**kwargs):
         choices=["pt", "lora", None, "none"],
         default="pt",
     )
-    parser.add_argument(
-        "--tracker",
-        type=str.lower,
-        choices=['aim', None, 'none'],
-        default="aim"
-    )
     (
         model_args,
         data_args,