Mlx FLUX.1 Schnell 4bit Quantized

🚀 DiffusionKit上的FLUX.1-schnell！

DiffusionKit MLX集成了FLUX.1-schnell模型，可用於文本到圖像的生成。它支持4位量化，能高效地根據文本提示生成高質量圖像。

🚀 快速開始

查看相關資源

• 查看原始模型！
• 查看DiffusionKit的GitHub倉庫！

模型效果展示

注意事項

本檢查點使用MLX的nn.quantize函數，以默認設置（group_size=64）對mmdit模塊進行了4位量化。

📦 安裝指南

創建conda環境

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

💻 使用示例

運行命令行工具

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

📄 許可證

本項目採用Apache-2.0許可證。