Upload 2 files

app.py

@@ -111,9 +111,11 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
     progress(1, desc="Preparing Inference.")
 
     image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress)
-    pipe.to("cpu")
+    if is_valid_lora(lora_json):
+        pipe.unfuse_lora()
+        pipe.unload_lora_weights()
     if selected_index is not None: pipe.unload_lora_weights()
-    if is_valid_lora(lora_json): pipe.unfuse_lora()
+    pipe.to("cpu")
     clear_cache()
     return image, seed  
 

mod.py

@@ -135,7 +135,7 @@ def fuse_loras(pipe, lorajson: list[dict]):
     if not a_list: return
     pipe.set_adapters(a_list, adapter_weights=w_list)
     pipe.fuse_lora(adapter_names=a_list, lora_scale=1.0)
-    pipe.unload_lora_weights()
+    #pipe.unload_lora_weights()
 
 
 change_base_model.zerogpu = True