Commit
•
51ec72f
1
Parent(s):
844263c
Update README.md
Browse files
README.md
CHANGED
|
@@ -22,7 +22,7 @@ You can use this with the 🧨Diffusers library from [Hugging Face](https://hugg
|
|
| 22 |
from diffusers import DiffusionPipeline
|
| 23 |
|
| 24 |
pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats")
|
| 25 |
-
pipe =
|
| 26 |
|
| 27 |
image = pipe().images[0]
|
| 28 |
|
|
|
|
| 22 |
from diffusers import DiffusionPipeline
|
| 23 |
|
| 24 |
pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats")
|
| 25 |
+
pipe = pipeline.to("cuda")
|
| 26 |
|
| 27 |
image = pipe().images[0]
|
| 28 |
|