Update README.md
#19
by
matospiso
- opened
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 |
|