Update app.py

app.py

@@ -116,9 +116,9 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
   pipe.to("cuda") 
   start_time = time()
   print("Loading LoRA weights...")
-  pipe.load_lora_weights(state_dict_1)
+  pipe.load_lora_weights(state_dict_1, low_cpu_mem_usage=True)
   pipe.fuse_lora(lora_1_scale)
-  pipe.load_lora_weights(state_dict_2)
+  pipe.load_lora_weights(state_dict_2, low_cpu_mem_usage=True)
   pipe.fuse_lora(lora_2_scale)
   lora_time = time() - start_time
   print(f"Loaded LoRAs time: {lora_time}")