Bump aim version.

Fix duplicate argument

Signed-off-by: Dushyant Behl <[email protected]>

Author: dushyantbehl

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

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"}
     )

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,