Upload folder using huggingface_hub

generate.py

@@ -9,7 +9,7 @@ NEGATIVE_PROMPT = (
 )
 
 LORA = "out/lora/realistic_aesthetic_v7.safetensors"
-PROMPTS_FILE = "out/prompts2.txt"
+PROMPTS_FILE = "out/prompts.txt"
 OUT_DIR = "out/img"
 
 
@@ -24,11 +24,16 @@ def load_pipe():
 
 def generate(pipe, prompt, path, lora_path=None, lora_scale=0.0):
     if lora_path:
-        if getattr(pipe, "_lora_loaded", False):
-            pipe.unload_lora_weights()
         pipe.load_lora_weights(lora_path)
-        pipe._lora_loaded = True
         pipe.fuse_lora(lora_scale=lora_scale)
+    else:
+        # ensure no LoRA is active
+        if hasattr(pipe, 'unfuse_lora'):
+            try: pipe.unfuse_lora()
+            except: pass
+        if hasattr(pipe, 'unload_lora_weights'):
+            try: pipe.unload_lora_weights()
+            except: pass
 
     image = pipe(
         prompt=prompt,