From 234158845c049dd547cd6fca39b072fa6297d9b9 Mon Sep 17 00:00:00 2001 From: DeepBeepMeep Date: Tue, 18 Mar 2025 00:36:56 +0100 Subject: [PATCH] Added check for already download loras --- gradio_server.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/gradio_server.py b/gradio_server.py index 1d07ea7..db5cc77 100644 --- a/gradio_server.py +++ b/gradio_server.py @@ -1486,15 +1486,24 @@ def download_loras(): yield "Please wait while the Loras are being downloaded", *[gr.Column(visible=False)] * 2 log_path = os.path.join(lora_dir, "log.txt") - if not os.path.isfile(log_path): + if not os.path.isfile(log_path) or True: import shutil tmp_path = os.path.join(lora_dir, "tmp_lora_dowload") import shutil, glob snapshot_download(repo_id="DeepBeepMeep/Wan2.1", allow_patterns="loras_i2v/*", local_dir= tmp_path) - [shutil.move(f, lora_dir) for f in glob.glob(os.path.join(tmp_path, "loras_i2v", "*.*")) if not "README.txt" in f ] - + for f in glob.glob(os.path.join(tmp_path, "loras_i2v", "*.*")): + target_file = os.path.join(lora_dir, Path(f).parts[-1] ) + if os.path.isfile(target_file): + os.remove(f) + else: + shutil.move(f, lora_dir) + try: + os.remove(tmp_path) + except: + pass + yield "Loras have been completely downloaded", *[gr.Column(visible=True)] * 2 from datetime import datetime