Update README.md
Browse files
README.md
CHANGED
@@ -57,13 +57,13 @@ pip install diffusers
|
|
57 |
|
58 |
```python
|
59 |
import torch
|
60 |
-
from diffusers import
|
61 |
|
62 |
device = "cuda"
|
63 |
dtype = torch.bfloat16
|
64 |
model = "Disty0/sotediffusion-wuerstchen3-alpha1-decoder"
|
65 |
|
66 |
-
pipe =
|
67 |
|
68 |
# send everything to the gpu:
|
69 |
pipe = pipe.to(device, dtype=dtype)
|
|
|
57 |
|
58 |
```python
|
59 |
import torch
|
60 |
+
from diffusers import StableCascadeCombinedPipeline
|
61 |
|
62 |
device = "cuda"
|
63 |
dtype = torch.bfloat16
|
64 |
model = "Disty0/sotediffusion-wuerstchen3-alpha1-decoder"
|
65 |
|
66 |
+
pipe = StableCascadeCombinedPipeline.from_pretrained(model, torch_dtype=dtype)
|
67 |
|
68 |
# send everything to the gpu:
|
69 |
pipe = pipe.to(device, dtype=dtype)
|