patrickvonplaten matospiso commited on
Commit
1beb956
1 Parent(s): fbda352

Update README.md (#19)

Browse files

- Update README.md (5266a6c6f889bbd527dab06fc1d3b1b1ccef7a0b)


Co-authored-by: Martin <[email protected]>

Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -97,6 +97,7 @@ 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
 
101
  init_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png").resize((512, 512))
102
 
 
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")
101
 
102
  init_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png").resize((512, 512))
103