Update app.py

app.py

@@ -38,7 +38,7 @@ for item in sdxl_loras:
         state_dict = torch.load(saved_name)
         
     item["saved_name"] = saved_name
-    item["state_dict"] = copy.deepcopy(state_dict) #{k: v.to(device="cuda", dtype=torch.float16) for k, v in state_dict.items() if torch.is_tensor(v)}
+    item["state_dict"] = state_dict #{k: v.to(device="cuda", dtype=torch.float16) for k, v in state_dict.items() if torch.is_tensor(v)}
 
 css = '''
 #title{text-align:center;}
@@ -58,8 +58,8 @@ css = '''
 original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
 
 def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lora_2_scale=0.5, progress=gr.Progress(track_tqdm=True)):
-  state_dict_1 = shuffled_items[0]['state_dict']
-  state_dict_2 = shuffled_items[1]['state_dict']
+  state_dict_1 = copy.deepcopy(shuffled_items[0]['state_dict'])
+  state_dict_2 = copy.deepcopy(shuffled_items[1]['state_dict'])
   pipe = copy.deepcopy(original_pipe)
   pipe.to("cuda")