fix: remove fire.fire(main), replace with main()

Signed-off-by: Will Johnson <[email protected]>
foundation-model-stack · Aug 30, 2024 · 7e80863 · 7e80863
1 parent 474e539
commit 7e80863
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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()