Mlx Stable Diffusion 3.5 Large 4bit Quantized

🚀 穩定擴散3.5大模型在DiffusionKit MLX上的量化版本！

本項目通過對MMDiT所有線性層應用nn.quantize生成檢查點。它主要用於文本到圖像的生成，藉助DiffusionKit MLX實現高效的圖像生成功能。

🚀 快速開始

環境準備

創建並激活Conda環境，同時安裝diffusionkit：

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 'SD3 on \
DiffusionKit' glowing softly. High detail hard surface render" \
--model-version argmaxinc/mlx-stable-diffusion-3.5-large-4bit-quantized \
--height 768 \
--width 1360 \
--seed 1001 \
--step 50 \
--cfg 7 \
--t5 \
--output ~/Desktop/sd3_on_mac.png

📄 許可證

本項目使用的許可證為stabilityai-ai-community。

🔗 相關鏈接

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

🖼️ 示例輸出