Update README.md
Browse files
README.md
CHANGED
|
@@ -20,8 +20,9 @@ pip install -U diffusers
|
|
| 20 |
and then you can run
|
| 21 |
```py
|
| 22 |
import torch
|
| 23 |
-
from diffusers import
|
| 24 |
-
|
|
|
|
| 25 |
pipe = pipe.to("cuda")
|
| 26 |
image = pipe(
|
| 27 |
"A capybara holding a sign that reads Hello World",
|
|
|
|
| 20 |
and then you can run
|
| 21 |
```py
|
| 22 |
import torch
|
| 23 |
+
from diffusers import DiffusionPipeline
|
| 24 |
+
|
| 25 |
+
pipe = DiffusionPipeline.from_pretrained("yandex/stable-diffusion-2-1-alchemist", torch_dtype=torch.float16)
|
| 26 |
pipe = pipe.to("cuda")
|
| 27 |
image = pipe(
|
| 28 |
"A capybara holding a sign that reads Hello World",
|