metadata

license: apache-2.0
prior:
  - warp-diffusion/wuerstchen-prior
tags:
  - text-to-image
  - wuerstchen

How-to-use

By using this pipeline with AutoPipelineForText2Image, the required prior pipeline: https://huggingface.co/warp-diffusion/wuerstchen-prior will automatically be downloaded and run.

import torch
from diffusers import AutoPipelineForText2Image
from diffusers.pipelines.wuerstchen import default_stage_c_timesteps

pipe = AutoPipelineForText2Image.from_pretrained("warp-ai/wuerstchen", torch_dtype=torch.float16).to("cuda")

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