remove unet

app.py

@@ -13,7 +13,7 @@ import os
 base = DiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0", 
     torch_dtype=torch.float16, 
-    variant="fp16", 
+    variant="fp16",
     use_safetensors=True
 )
 base.to("cuda:0")
@@ -30,13 +30,13 @@ pipe.to("cuda:0")
 refiner = DiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-refiner-1.0",
     text_encoder_2=base.text_encoder_2,
+    torch_dtype=torch.float16, 
     vae=base.vae,
-    torch_dtype=torch.float16,
     use_safetensors=True,
     variant="fp16",
 )
 refiner.to("cuda:0")
-refiner.unet = torch.compile(refiner.unet, mode="reduce-overhead", fullgraph=True)
+
 
 @spaces.GPU  # Apply the GPU decorator
 def generate_and_save_image(prompt, negative_prompt=''):