Geek7 commited on
Commit
1d08e64
·
verified ·
1 Parent(s): 419154d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from diffusers import DiffusionPipeline
2
  import torch
3
 
4
- pipeline = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch.float16)
5
  pipeline.to("cpu")
6
  pipeline("An image of a squirrel in Picasso style").images[0]
 
1
  from diffusers import DiffusionPipeline
2
  import torch
3
 
4
+ pipeline = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch.float32)
5
  pipeline.to("cpu")
6
  pipeline("An image of a squirrel in Picasso style").images[0]