Simplify Wuerstchen model card

#1
Files changed (1) hide show
  1. README.md +7 -21
README.md CHANGED
@@ -35,39 +35,25 @@ This pipeline should be run together with a prior https://huggingface.co/warp-ai
35
 
36
  ```py
37
  import torch
38
- from diffusers import WuerstchenDecoderPipeline, WuerstchenPriorPipeline
39
- from diffusers.pipelines.wuerstchen import default_stage_c_timesteps
40
 
41
  device = "cuda"
42
  dtype = torch.float16
43
- num_images_per_prompt = 2
44
 
45
- prior_pipeline = WuerstchenPriorPipeline.from_pretrained(
46
- "warp-ai/wuerstchen-prior", torch_dtype=dtype
47
- ).to(device)
48
- decoder_pipeline = WuerstchenDecoderPipeline.from_pretrained(
49
- "warp-ai/wuerstchen", torch_dtype=dtype
50
  ).to(device)
51
 
52
  caption = "Anthropomorphic cat dressed as a fire fighter"
53
  negative_prompt = ""
54
 
55
- prior_output = prior_pipeline(
56
  prompt=caption,
57
  height=1024,
58
- width=1536,
59
- timesteps=default_stage_c_timesteps,
60
- negative_prompt=negative_prompt,
61
- guidance_scale=4.0,
62
- num_images_per_prompt=num_images_per_prompt,
63
- )
64
- decoder_output = decoder_pipeline(
65
- image_embeddings=prior_output.image_embeddings,
66
- prompt=caption,
67
  negative_prompt=negative_prompt,
68
- num_images_per_prompt=num_images_per_prompt,
69
- guidance_scale=0.0,
70
- output_type="pil",
71
  ).images
72
  ```
73
 
 
35
 
36
  ```py
37
  import torch
38
+ from diffusers import AutoPipelineForText2Image
 
39
 
40
  device = "cuda"
41
  dtype = torch.float16
 
42
 
43
+ pipeline = AutoPipelineForText2Image.from_pretrained(
44
+ "warp-diffusion/wuerstchen", torch_dtype=dtype
 
 
 
45
  ).to(device)
46
 
47
  caption = "Anthropomorphic cat dressed as a fire fighter"
48
  negative_prompt = ""
49
 
50
+ output = pipeline(
51
  prompt=caption,
52
  height=1024,
53
+ width=1024,
 
 
 
 
 
 
 
 
54
  negative_prompt=negative_prompt,
55
+ prior_guidance_scale=4.0,
56
+ decoder_guidance_scale=0.0,
 
57
  ).images
58
  ```
59