Update README.md

README.md

@@ -34,33 +34,13 @@ pip install diffusers
 
 ```python
 import torch
-from diffusers import StableCascadeCombinedPipeline, StableCascadePriorPipeline, StableCascadeDecoderPipeline
-
-
+from diffusers import StableCascadeCombinedPipeline
 
 device = "cuda"
 dtype = torch.bfloat16
+model = "Disty0/sotediffusion-wuerstchen3-alpha1-decoder"
 
-prior_model = "Disty0/sotediffusion-wuerstchen3-alpha1"
-decoder_model = "Disty0/sotediffusion-wuerstchen3-alpha1-decoder"
-
-
-prior = StableCascadePriorPipeline.from_pretrained(prior_model, torch_dtype=dtype)
-decoder = StableCascadeDecoderPipeline.from_pretrained(decoder_model, torch_dtype=dtype)
-
-pipe = StableCascadeCombinedPipeline(
-    tokenizer=decoder.tokenizer,
-    text_encoder=decoder.text_encoder,
-    decoder=decoder.decoder,
-    scheduler=decoder.scheduler,
-    vqgan=decoder.vqgan,
-    prior_prior=prior.prior,
-    prior_text_encoder=prior.text_encoder,
-    prior_tokenizer=prior.tokenizer,
-    prior_scheduler=prior.scheduler,
-    prior_feature_extractor=prior.feature_extractor,
-    prior_image_encoder=prior.image_encoder)
-
+pipe = StableCascadeCombinedPipeline.from_pretrained(model, torch_dtype=dtype)
 
 # send everything to the gpu:
 pipe = pipe.to(device, dtype=dtype)
@@ -87,6 +67,7 @@ output = pipe(
 ).images[0]
 
 ## do something with the output image
+
 ```