Update README.md

README.md

@@ -15,15 +15,17 @@ and run.
 ```py
 import torch
 from diffusers import AutoPipelineForText2Image
+from diffusers.pipelines.wuerstchen import default_stage_c_timesteps
 
-pipe = AutoPipelineForText2Image.from_pretrained("warp-diffusion/wuerstchen", torch_dtype=torch.float16).to("cuda")
+pipe = AutoPipelineForText2Image.from_pretrained("warp-ai/wuerstchen", torch_dtype=torch.float16).to("cuda")
 
-prompt = [
-    "An old destroyed car standing on a cliff in norway, cinematic photography",
-    "Western movie, closeup cinematic photography",
-    "Pink nike shoe commercial, closeup cinematic photography",
-    "Croatia, closeup cinematic photography",
-    "South Tyrol mountains at sunset, closeup cinematic photography",
-]
-images = pipe(prompt, guidance_scale=8.0, width=1024, height=1024).images
+caption = "Anthropomorphic cat dressed as a fire fighter"
+images = pipe(
+    caption, 
+    width=1024,
+    height=1536,
+    prior_timesteps=default_stage_c_timesteps,
+    prior_guidance_scale=4.0,
+    num_images_per_prompt=2,
+).images
 ```