Browse Source

Add llama_finetuning.py script to provide support for torchrun

Matthias Reso 1 year ago
parent
commit
27e56bdfd3
1 changed files with 8 additions and 0 deletions
  1. 8 0
      llama_finetuning.py

+ 8 - 0
llama_finetuning.py

@@ -0,0 +1,8 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# This software may be used and distributed according to the terms of the Llama 2 Community License Agreement.
+
+import fire
+from llama_recipes.finetuning import main
+
+if __name__ == "__main__":
+    fire.Fire(main)