alidev2002 commited on
Commit
e91e2a1
Β·
verified Β·
1 Parent(s): aac4eb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
 
5
  pipe = DiffusionPipeline.from_pretrained(
6
  "stabilityai/stable-diffusion-xl-base-1.0",
7
- torch_dtype=torch.float16,
8
  use_safetensors=True,
9
  variant="fp16"
10
  )
 
4
 
5
  pipe = DiffusionPipeline.from_pretrained(
6
  "stabilityai/stable-diffusion-xl-base-1.0",
7
+ torch_dtype=torch.float32,
8
  use_safetensors=True,
9
  variant="fp16"
10
  )