patrickvonplaten
commited on
Merge branch 'main' of https://huggingface.co/stabilityai/stable-diffusion-2-1-unclip-small into main
Browse files
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
|
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
|