Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from time import sleep
+from time import sleep, time
 from diffusers import DiffusionPipeline
 from huggingface_hub import hf_hub_download, CommitScheduler
 from safetensors.torch import load_file
@@ -12,6 +12,7 @@ import json
 import random
 import copy
 import gc
+import pickle
 import spaces
 
 lora_list = hf_hub_download(repo_id="multimodalart/LoraTheExplorer", filename="sdxl_loras.json", repo_type="space")
@@ -94,10 +95,15 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
   repo_id_1 = shuffled_items[0]['repo']
   repo_id_2 = shuffled_items[1]['repo']
   print("Loading state dicts...")
-  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_1 = state_dicts[repo_id_1]["state_dict"]
+  state_dict_2 = state_dicts[repo_id_2]["state_dict"]
   print("Loaded state dicts.")
-  pipe = copy.deepcopy(original_pipe)
+  #pipe = copy.deepcopy(original_pipe)
+  # Time for pickle
+  start_time = time.time()
+  pipe = pickle.loads(pickle.dumps(original_pipe))
+  pickle_time = time.time() - start_time
+  print(f"Pickle time: {pickle_time}")  
   pipe.to("cuda") 
   print("Loading LoRA weights...")
   pipe.load_lora_weights(state_dict_1)