ソースを参照

Add llama_finetuning.py script to provide support for torchrun

Matthias Reso 1 年間 前
コミット
27e56bdfd3
1 ファイル変更8 行追加0 行削除
  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)