candle-wuerstchen: Efficient Pretraining of Text-to-Image Models

The wuerstchen example is a port of the diffusers implementation for Würstchen v2. The candle implementation reproduces the same structure/files for models and pipelines. Useful resources:

• Official implementation.
• Arxiv paper.
• Blog post: Introducing Würstchen: Fast Diffusion for Image Generation.

Getting the weights

The weights are automatically downloaded for you from the HuggingFace Hub on the first run. There are various command line flags to use local files instead, run with --help to learn about them.

Running some example.

cargo run --example wuerstchen --release --features cuda,cudnn -- \
  --prompt "Anthropomorphic cat dressed as a fire fighter"

The final image is named sd_final.png by default.