mflux_steps.md

August 16, 2024 ยท View on GitHub

Setup the repo

git clone git@github.com:filipstrand/mflux.git
cd mflux && pip install -r requirements.txt

Make a run script

Name this anything, maybe flux.py. Make sure to update the two paths marked below.

import sys
sys.path.append("/path/to/mflux/repo/src")  # <- Update

from flux_1_schnell.config.config import Config
from flux_1_schnell.models.flux import Flux1Schnell

flux = Flux1Schnell("black-forest-labs/FLUX.1-schnell")

image = flux.generate_image(
    seed=3,
    prompt="A rocketship exactly in the style of the cartoon rocket emoji. The rocket ship is flying in the sky. The logo 'MLX' is written on the side of the ship in black letters.",
    config=Config(num_inference_steps=2)
)

image.save("/path/to/save/image.png")  # <- Update

Run the script

python flux.py

NB The model weights will be downloaded on the first run. This can take a long time.