hongming commited on
Commit
2981f10
·
1 Parent(s): 7107a1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ st.write(x, 'squared is', x * x)
6
  from diffusers import DiffusionPipeline
7
  import torch
8
 
9
- pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
10
 
11
 
12
  # if using torch < 2.0
 
6
  from diffusers import DiffusionPipeline
7
  import torch
8
 
9
+ pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float32, use_safetensors=True, variant="fp32")
10
 
11
 
12
  # if using torch < 2.0