Commit
•
1beb956
1
Parent(s):
fbda352
Update README.md (#19)
Browse files- Update README.md (5266a6c6f889bbd527dab06fc1d3b1b1ccef7a0b)
Co-authored-by: Martin <[email protected]>
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 |
|