Update app.py

app.py

@@ -96,15 +96,15 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
   repo_id_2 = shuffled_items[1]['repo']
   print("Loading state dicts...")
   start_time = time()  
-  state_dict_1 = pickle.loads(pickle.dumps(state_dicts[repo_id_1]["state_dict"]))
-  state_dict_2 = pickle.loads(pickle.dumps(state_dicts[repo_id_2]["state_dict"]))
+  state_dict_1 = copy.deepcopy(state_dicts[repo_id_1]["state_dict"])
+  state_dict_2 = copy.deepcopy(state_dicts[repo_id_2]["state_dict"])
   state_dict_time = time() - start_time
   print(f"State Dict time: {state_dict_time}")  
   #pipe = copy.deepcopy(original_pipe)
   start_time = time()
-  pipe = pickle.loads(pickle.dumps(original_pipe))
+  pipe.unet = copy.deepcopy(original_pipe.unet)
   pickle_time = time() - start_time
-  print(f"Pickle time: {pickle_time}")  
+  print(f"copy time: {pickle_time}")  
   pipe.to("cuda") 
   start_time = time()
   print("Loading LoRA weights...")