Simplify Wuerstchen model card

README.md

@@ -35,39 +35,25 @@ This pipeline should be run together with a prior https://huggingface.co/warp-ai
 
 ```py
 import torch
-from diffusers import WuerstchenDecoderPipeline, WuerstchenPriorPipeline
-from diffusers.pipelines.wuerstchen import default_stage_c_timesteps
+from diffusers import AutoPipelineForText2Image
 
 device = "cuda"
 dtype = torch.float16
-num_images_per_prompt = 2
 
-prior_pipeline = WuerstchenPriorPipeline.from_pretrained(
-    "warp-ai/wuerstchen-prior", torch_dtype=dtype
-).to(device)
-decoder_pipeline = WuerstchenDecoderPipeline.from_pretrained(
-    "warp-ai/wuerstchen", torch_dtype=dtype
+pipeline =  AutoPipelineForText2Image.from_pretrained(
+    "warp-diffusion/wuerstchen", torch_dtype=dtype
 ).to(device)
 
 caption = "Anthropomorphic cat dressed as a fire fighter"
 negative_prompt = ""
 
-prior_output = prior_pipeline(
+output = pipeline(
     prompt=caption,
     height=1024,
-    width=1536,
-    timesteps=default_stage_c_timesteps,
-    negative_prompt=negative_prompt,
-	guidance_scale=4.0,
-    num_images_per_prompt=num_images_per_prompt,
-)
-decoder_output = decoder_pipeline(
-    image_embeddings=prior_output.image_embeddings,
-    prompt=caption,
+    width=1024,
     negative_prompt=negative_prompt,
-    num_images_per_prompt=num_images_per_prompt,
-    guidance_scale=0.0,
-    output_type="pil",
+    prior_guidance_scale=4.0,
+    decoder_guidance_scale=0.0,
 ).images
 ```