Disty0 commited on
Commit
2ee80ee
1 Parent(s): b43813e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -57,13 +57,13 @@ pip install diffusers
57
 
58
  ```python
59
  import torch
60
- from diffusers import AutoPipelineForText2Image
61
 
62
  device = "cuda"
63
  dtype = torch.bfloat16
64
  model = "Disty0/sotediffusion-wuerstchen3-alpha1-decoder"
65
 
66
- pipe = AutoPipelineForText2Image.from_pretrained(model, torch_dtype=dtype)
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)