Hugging Face's logo

warp-ai
/
wuerstchen-prior

Diffusers
Safetensors
WuerstchenPriorPipeline
Model card Files Community
kashif HF staff commited on
Commit
3075d30
•
1 Parent(s): ced6b20

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -27,7 +27,7 @@ This pipeline should be run together with https://huggingface.co/warp-ai/wuerstc
27
  ```py
28
  import torch
29
  from diffusers import WuerstchenDecoderPipeline, WuerstchenPriorPipeline
30
- from diffusers.pipelines.wuerstchen import default_stage_c_timesteps
31
 
32
  device = "cuda"
33
  dtype = torch.float16
@@ -47,7 +47,7 @@ prior_output = prior_pipeline(
47
  prompt=caption,
48
  height=1024,
49
  width=1536,
50
- timesteps=default_stage_c_timesteps,
51
  negative_prompt=negative_prompt,
52
  guidance_scale=4.0,
53
  num_images_per_prompt=num_images_per_prompt,
@@ -56,7 +56,6 @@ decoder_output = decoder_pipeline(
56
  image_embeddings=prior_output.image_embeddings,
57
  prompt=caption,
58
  negative_prompt=negative_prompt,
59
- num_images_per_prompt=num_images_per_prompt,
60
  guidance_scale=0.0,
61
  output_type="pil",
62
  ).images
 
27
  ```py
28
  import torch
29
  from diffusers import WuerstchenDecoderPipeline, WuerstchenPriorPipeline
30
+ from diffusers.pipelines.wuerstchen import DEFAULT_STAGE_C_TIMESTEPS
31
 
32
  device = "cuda"
33
  dtype = torch.float16
 
47
  prompt=caption,
48
  height=1024,
49
  width=1536,
50
+ timesteps=DEFAULT_STAGE_C_TIMESTEPS,
51
  negative_prompt=negative_prompt,
52
  guidance_scale=4.0,
53
  num_images_per_prompt=num_images_per_prompt,
 
56
  image_embeddings=prior_output.image_embeddings,
57
  prompt=caption,
58
  negative_prompt=negative_prompt,
 
59
  guidance_scale=0.0,
60
  output_type="pil",
61
  ).images