Browse Source

disable nightlies check

Hamid Shojanazeri 1 năm trước cách đây
mục cha
commit
85bcea1765
1 tập tin đã thay đổi với 3 bổ sung3 xóa
  1. 3 3
      src/llama_recipes/finetuning.py

+ 3 - 3
src/llama_recipes/finetuning.py

@@ -76,9 +76,9 @@ def main(**kwargs):
         """
         """
         v = packaging.version.parse(torch.__version__)
         v = packaging.version.parse(torch.__version__)
         verify_latest_nightly = v.is_devrelease and v.dev >= 20230701
         verify_latest_nightly = v.is_devrelease and v.dev >= 20230701
-        if not verify_latest_nightly:
-            raise Exception("latest pytorch nightly build is required to run with low_cpu_fsdp config, "
-                            "please install latest nightly.")
+        #if not verify_latest_nightly:
+            #raise Exception("latest pytorch nightly build is required to run with low_cpu_fsdp config, "
+                            #"please install latest nightly.")
         if rank == 0:
         if rank == 0:
             model = LlamaForCausalLM.from_pretrained(
             model = LlamaForCausalLM.from_pretrained(
                 train_config.model_name,
                 train_config.model_name,