IDKiro commited on
Commit
f8ee0c1
1 Parent(s): d3bc7ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ weight_type = torch.float16
15
  pipe = StableDiffusionPipeline.from_pretrained("IDKiro/sdxs-512-dreamshaper")
16
  pipe.unet = PeftModel.from_pretrained(pipe.unet, "IDKiro/sdxs-512-dreamshaper-anime")
17
  pipe.unet.merge_and_unload()
18
- pipe.to(torch_device=device, torch_dtype=weight_type)
19
 
20
  vae_tiny = AutoencoderTiny.from_pretrained(
21
  "IDKiro/sdxs-512-dreamshaper", subfolder="vae"
 
15
  pipe = StableDiffusionPipeline.from_pretrained("IDKiro/sdxs-512-dreamshaper")
16
  pipe.unet = PeftModel.from_pretrained(pipe.unet, "IDKiro/sdxs-512-dreamshaper-anime")
17
  pipe.unet.merge_and_unload()
18
+ pipe.to(device, dtype=weight_type)
19
 
20
  vae_tiny = AutoencoderTiny.from_pretrained(
21
  "IDKiro/sdxs-512-dreamshaper", subfolder="vae"