소스 검색

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)