patrickvonplaten vikasp commited on
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>

Files changed (1) hide show
  1. README.md +1 -0
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")