diff --git a/tuning/sft_trainer.py b/tuning/sft_trainer.py index b5e6cb62e..d5ff61aaf 100644 --- a/tuning/sft_trainer.py +++ b/tuning/sft_trainer.py @@ -37,7 +37,6 @@ ) from transformers.utils import is_accelerate_available from trl import SFTConfig, SFTTrainer -import fire import transformers # Local @@ -629,4 +628,4 @@ def main(**kwargs): # pylint: disable=unused-argument if __name__ == "__main__": - fire.Fire(main) + main()