Hugging Face's logo

warp-ai
/
wuerstchen-prior-model-interpolated

Diffusers
Safetensors
Model card Files Community
kashif HF staff commited on
Commit
4e8adf7
•
1 Parent(s): 55ecd3b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -34,7 +34,7 @@ This pipeline should be run together with https://huggingface.co/warp-ai/wuerstc
34
  ```py
35
  import torch
36
  from diffusers import WuerstchenDecoderPipeline, WuerstchenPriorPipeline
37
- from diffusers.pipelines.wuerstchen import default_stage_c_timesteps
38
 
39
  device = "cuda"
40
  dtype = torch.float16
@@ -54,7 +54,7 @@ prior_output = prior_pipeline(
54
  prompt=caption,
55
  height=1024,
56
  width=1536,
57
- timesteps=default_stage_c_timesteps,
58
  negative_prompt=negative_prompt,
59
  guidance_scale=4.0,
60
  num_images_per_prompt=num_images_per_prompt,
@@ -63,7 +63,6 @@ decoder_output = decoder_pipeline(
63
  image_embeddings=prior_output.image_embeddings,
64
  prompt=caption,
65
  negative_prompt=negative_prompt,
66
- num_images_per_prompt=num_images_per_prompt,
67
  guidance_scale=0.0,
68
  output_type="pil",
69
  ).images
 
34
  ```py
35
  import torch
36
  from diffusers import WuerstchenDecoderPipeline, WuerstchenPriorPipeline
37
+ from diffusers.pipelines.wuerstchen import DEFAULT_STAGE_C_TIMESTEPS
38
 
39
  device = "cuda"
40
  dtype = torch.float16
 
54
  prompt=caption,
55
  height=1024,
56
  width=1536,
57
+ timesteps=DEFAULT_STAGE_C_TIMESTEPS,
58
  negative_prompt=negative_prompt,
59
  guidance_scale=4.0,
60
  num_images_per_prompt=num_images_per_prompt,
 
63
  image_embeddings=prior_output.image_embeddings,
64
  prompt=caption,
65
  negative_prompt=negative_prompt,
 
66
  guidance_scale=0.0,
67
  output_type="pil",
68
  ).images