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.