Update README.md
Browse files
README.md
CHANGED
@@ -73,10 +73,10 @@ prior_model = "Disty0/sotediffusion-wuerstchen3-alpha1"
|
|
73 |
decoder_model = "Disty0/sotediffusion-wuerstchen3-alpha1-decoder"
|
74 |
|
75 |
|
76 |
-
prior =
|
77 |
-
decoder =
|
78 |
|
79 |
-
pipe =
|
80 |
tokenizer=decoder.tokenizer,
|
81 |
text_encoder=decoder.text_encoder,
|
82 |
decoder=decoder.decoder,
|
|
|
73 |
decoder_model = "Disty0/sotediffusion-wuerstchen3-alpha1-decoder"
|
74 |
|
75 |
|
76 |
+
prior = StableCascadePriorPipeline.from_pretrained(prior_model, torch_dtype=dtype)
|
77 |
+
decoder = StableCascadeDecoderPipeline.from_pretrained(decoder_model, torch_dtype=dtype)
|
78 |
|
79 |
+
pipe = StableCascadeCombinedPipeline(
|
80 |
tokenizer=decoder.tokenizer,
|
81 |
text_encoder=decoder.text_encoder,
|
82 |
decoder=decoder.decoder,
|