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