Don't delete weights path

replicate · Oct 11, 2024 · 3285317 · 3285317
1 parent 9359a41
commit 3285317
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/train.py b/train.py
@@ -474,9 +474,6 @@ def clean_up():
     if OUTPUT_DIR.exists():
         shutil.rmtree(OUTPUT_DIR)
 
-    if WEIGHTS_PATH.exists():
-        shutil.rmtree(WEIGHTS_PATH)
-
 
 def download_huggingface_lora(hf_lora_url: str, output_path: str):
     if (