Upload app.py

app.py

@@ -78,8 +78,8 @@ def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height,
 
 def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height,
               lora_scale, lora_repo, lora_weights, lora_trigger, progress=gr.Progress(track_tqdm=True)):
-    if selected_index is None and not lora_repo:
-        raise gr.Error("You must select a LoRA before proceeding.")
+    #if selected_index is None and not lora_repo:
+    #    raise gr.Error("You must select a LoRA before proceeding.")
 
     if selected_index is not None and not lora_repo:
         selected_lora = loras[selected_index]
@@ -92,7 +92,9 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
 
     # Load LoRA weights
     with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"):
-        if lora_weights: # override
+        if selected_index is None and not lora_repo: # override
+            pass
+        elif lora_weights: # override
             pipe.load_lora_weights(lora_path, weight_name=lora_weights)
         elif "weights" in selected_lora:
             pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"])
@@ -106,7 +108,7 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
     
     image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress)
     pipe.to("cpu")
-    pipe.unload_lora_weights()
+    if selected_index is not None or lora_repo: pipe.unload_lora_weights()
     return image, seed  
 
 run_lora.zerogpu = True