Update app.py

app.py

@@ -344,16 +344,13 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
             lora_path = lora['repo']
             weight_name = lora.get("weights")
             print(f"Lora Path: {lora_path}")
-            if image_input is not None:
-                if weight_name:
-                    pipe_i2i.load_lora_weights(lora_path, weight_name=weight_name, low_cpu_mem_usage=True, adapter_name=lora_name)
-                else:
-                    pipe_i2i.load_lora_weights(lora_path, low_cpu_mem_usage=True, adapter_name=lora_name)
-            else:
-                if weight_name:
-                    pipe.load_lora_weights(lora_path, weight_name=weight_name, low_cpu_mem_usage=True, adapter_name=lora_name)
-                else:
-                    pipe.load_lora_weights(lora_path, low_cpu_mem_usage=True, adapter_name=lora_name)
+            pipe_to_use = pipe_i2i if image_input is not None else pipe
+            pipe_to_use.load_lora_weights(
+                lora_path, 
+                weight_name=weight_name if weight_name else None,
+                low_cpu_mem_usage=False,
+                adapter_name=lora_name
+            )
         print("Loaded LoRAs:", lora_names)
         print("Adapter weights:", lora_weights)
         if image_input is not None: