From 3285317979f9d39261a3f3e4aa162b315689de3a Mon Sep 17 00:00:00 2001
From: andreasjansson <andreas@replicate.ai>
Date: Fri, 11 Oct 2024 15:15:58 +0200
Subject: [PATCH] Don't delete weights path

---
 train.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/train.py b/train.py
index ef08cc7..56ca767 100644
--- 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 (