Update app.py

app.py

@@ -15,7 +15,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)
+pipe.to(device, dtype=weight_type)
 
 vae_tiny = AutoencoderTiny.from_pretrained(
     "IDKiro/sdxs-512-dreamshaper", subfolder="vae"