diff --git a/requirements.txt b/requirements.txt index c4e8fa1ef..8f658ca25 100644 --- 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 index 7919841c0..10a024e6d 100644 --- 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 index 75d335bd9..c9c82303b 100644 --- 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,