FLUX.1-schnell on DiffusionKit MLX!

Check out the original model!

Check out the DiffusionKit github repository!

Note: This checkpoint features 4-bit quantization of the mmdit module using MLX's nn.quantize function with default settings (group_size=64).

Usage

• Create conda environment

conda create -n diffusionkit python=3.11 -y
conda activate diffusionkit
pip install diffusionkit

• Run the cli command

diffusionkit-cli --prompt "detailed cinematic dof render of a \
detailed MacBook Pro on a wooden desk in a dim room with items \
around, messy dirty room. On the screen are the letters 'FLUX on \
DiffusionKit' glowing softly. High detail hard surface render" \
--model-version argmaxinc/mlx-FLUX.1-schnell-4bit-quantized \
--height 768 \
--width 1360 \
--seed 1001 \
--step 4 \
--output ~/Desktop/flux_on_mac.png