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