Pārlūkot izejas kodu

Add llama_finetuning.py script to provide support for torchrun

Matthias Reso 1 gadu atpakaļ
vecāks
revīzija
27e56bdfd3
1 mainītis faili ar 8 papildinājumiem un 0 dzēšanām
  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)