FLUX.1-schnell on DiffusionKit MLX!
Check out the original model!
Check out the DiffusionKit github repository!
![FLUX.1 [schnell] Grid](./flux_on_mac.png)
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