mflux_steps.md

Setup the repo

git clone [email protected]: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.

@awni Thanks a lot for the tip! This has now been updated in main.

@uogbuji Thanks for your provided example. I ended up going with what @awni suggested and used the snapshot_download feature. This also mirrors how a typical Huggingface Diffusers setup work, so I tried to stay as close to that as possible. If you want to have a look at the changes, they can be found in this commit: filipstrand/mflux@e3a85df

Love it!!

I'm adding M4 Max results: 18.7 seconds!

mflux-generate --prompt "Luxury food photograph" --model schnell --steps 2 2 1.75s user 10.71s system 66% cpu 18.721 total

@ivanfioravanti Added to the performance table!