candle-flux: image generation with latent rectified flow transformers

Flux is a 12B rectified flow transformer capable of generating images from text descriptions, huggingface, github, blog post.

Running the model

bash

cargo run --features cuda --example flux -r -- \
    --height 1024 --width 1024 \
    --prompt "a rusty robot walking on a beach holding a small torch, the robot has the word "rust" written on it, high quality, 4k"