Download

Local installation

cd $HOME && git clone https://github.com/black-forest-labs/flux

cd $HOME/flux

python3.10 -m venv .venv

source .venv/bin/activate

pip install -e '.[all]'

There are three models:

  • FLUX.1 [pro] the base model, available via API

  • FLUX.1 [dev] guidance-distilled variant

  • FLUX.1 [schnell] guidance and step-distilled variant

The weights of the autoencoder are also released under apache-2.0 and can be found in either of the two HuggingFace repos above. They are the same for both models.

The weights will be downloaded automatically from HuggingFace once you start one of the demos. To download FLUX.1 [dev], you will need to be logged in, see here. If you have downloaded the model weights manually, you can specify the downloaded paths via environment-variables:

export FLUX_SCHNELL=<path_to_flux_schnell_sft_file>

export FLUX_DEV=<path_to_flux_dev_sft_file>

export AE=<path_to_ae_sft_file>

For interactive sampling run

python -m flux --name <name> --loop