patrickvonplaten commited on
Commit
60a89b7
2 Parent(s): e87e0c2 ddb7684

Merge branch 'main' of https://huggingface.co/stabilityai/stable-diffusion-2-1-unclip-small into main

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -49,7 +49,7 @@ from io import BytesIO
49
 
50
  from diffusers import DiffusionPipeline
51
 
52
- pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-unclip-small", torch_dtype=torch.float16, variant="fp16")
53
  pipe.to("cuda")
54
 
55
  # get image
 
49
 
50
  from diffusers import DiffusionPipeline
51
 
52
+ pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-unclip-small", torch_dtype=torch.float16)
53
  pipe.to("cuda")
54
 
55
  # get image