Update app.py

app.py

@@ -14,6 +14,7 @@ weight_type = torch.float16
 
 pipe = StableDiffusionPipeline.from_pretrained("IDKiro/sdxs-512-dreamshaper")
 pipe.unet = PeftModel.from_pretrained(pipe.unet, "IDKiro/sdxs-512-dreamshaper-anime")
+pipe.unet.merge_and_unload()
 pipe.to(torch_device=device, torch_dtype=weight_type)
 
 vae_tiny = AutoencoderTiny.from_pretrained(