stack86 commited on
Commit
cb18ee7
1 Parent(s): 7311c4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ if torch.cuda.is_available():
20
  # pipe.vae = AutoencoderKL.from_pretrained("IDKiro/sdxs-512-0.9/vae_large") # use original VAE
21
  pipe.to("cuda")
22
  else:
23
- weight_type = torch.float16
24
  pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=weight_type)
25
 
26
  # Function
 
20
  # pipe.vae = AutoencoderKL.from_pretrained("IDKiro/sdxs-512-0.9/vae_large") # use original VAE
21
  pipe.to("cuda")
22
  else:
23
+ weight_type = torch.float32
24
  pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=weight_type)
25
 
26
  # Function