Commit
•
f4b0486
1
Parent(s):
1beb956
update the readme to fix the image-to-image example (#18)
Browse files- update the readme to fix the image-to-image example (90c16b105674c53d930b3df3d3d2f9471a0518f9)
Co-authored-by: Vikas <vikasp@users.noreply.huggingface.co>
README.md
CHANGED
@@ -95,6 +95,7 @@ below.
|
|
95 |
```py
|
96 |
from diffusers import AutoPipelineForImage2Image
|
97 |
from diffusers.utils import load_image
|
|
|
98 |
|
99 |
pipe = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
|
100 |
pipe.to("cuda")
|
|
|
95 |
```py
|
96 |
from diffusers import AutoPipelineForImage2Image
|
97 |
from diffusers.utils import load_image
|
98 |
+
import torch
|
99 |
|
100 |
pipe = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
|
101 |
pipe.to("cuda")
|