Skip to content

Latest commit

 

History

History
 
 

wuerstchen

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

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

anthropomorphic cat dressed as a fire fighter

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:

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.