Просмотр исходного кода

Add llama_finetuning.py script to provide support for torchrun

Matthias Reso 1 год назад
Родитель
Сommit
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)