Update app.py

app.py

@@ -35,14 +35,11 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     vae=vae,
     torch_dtype=torch.float16,
-).to("cpu")
-original_pipe = copy.deepcopy(pipe)
-pipe.to(device)
+).to(device)
 
 last_lora = ""
 last_merged = False
 
-
 def update_selection(selected_state: gr.SelectData):
     lora_repo = sdxl_loras[selected_state.index]["repo"]
     instance_prompt = sdxl_loras[selected_state.index]["trigger_word"]
@@ -132,9 +129,11 @@ def run_lora(prompt, negative, lora_scale, selected_state):
     cross_attention_kwargs = None
     if last_lora != repo_name:
         if last_merged:
-            del pipe
-            pipe = copy.deepcopy(original_pipe)
-            pipe.to(device)
+            pipe = StableDiffusionXLPipeline.from_pretrained(
+                "stabilityai/stable-diffusion-xl-base-1.0",
+                vae=vae,
+                torch_dtype=torch.float16,
+            ).to(device)
         else:
             pipe.unload_lora_weights()
         is_compatible = sdxl_loras[selected_state.index]["is_compatible"]