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-v1-5-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-v1-5-alchemist", torch_dtype=torch.float16)
|
24 |
pipe = pipe.to("cuda")
|
25 |
image = pipe(
|
26 |
"A capybara holding a sign that reads Hello World",
|