Spaces:

Kunbyte
/

DRA-Ctrl

Running on Zero

caohy666 commited on 6 days ago

Commit

399a4ec

1 Parent(s): b187df6

<fix> remove resume_download

Files changed (1) hide show

app.py CHANGED Viewed

@@ -172,8 +172,7 @@ def process_image_and_text(condition_image, target_prompt, condition_image_promp
         # load LoRA weights
         model_root = hf_hub_download(
             repo_id="Kunbyte/DRA-Ctrl",
-            filename=f"{task}.safetensors",
-            resume_download=True)
         try:
             with safe_open(model_root, framework="pt") as f:

         # load LoRA weights
         model_root = hf_hub_download(
             repo_id="Kunbyte/DRA-Ctrl",
+            filename=f"{task}.safetensors")
         try:
             with safe_open(model_root, framework="pt") as f: