Views
No views yet
Ported by Mrbizarro · MIT licensed · published to mlx-community
model.safetensors — Q8 quantized backbone (10 GB)extras/custom_heads.safetensors — diffusion-side heads (75 MB, BF16)config.json (with quantization: {bits: 8, group_size: 64} so mlx-vlm wraps Linear → QuantizedLinear correctly)scripts/hidream_o1/. Clone that for code, this for weights only.1# Get the code from the BF16 sibling
2hf download mlx-community/HiDream-O1-Image-Dev-mlx-bf16 --local-dir hidream-o1-mlx \
3 --include "scripts/*" --include "*.md" --include "*.txt" --include "*.gitattributes"
4cd hidream-o1-mlx
5uv venv --python 3.11 && uv pip install -r requirements.txt
6
7# Get the Q8 weights
8hf download mlx-community/HiDream-O1-Image-Dev-mlx-q8 --local-dir mlx_models/hidream-o1-dev-q8
9
10# Run (square dims only for clean output)
11.venv/bin/python scripts/hidream_o1/generate_hidream_o1_mlx.py \
12 --model-path mlx_models/hidream-o1-dev-q8 \
13 --prompt "your prompt here" \
14 --width 2048 --height 2048 \
15 --output out.png| Resolution | Per step | Total (28 steps) | Peak RAM | Quality |
|---|---|---|---|---|
| 1024×1024 | 2.36 s | 67 s | 11.5 GB | ✅ clean |
| 2048×2048 | 6.68 s | 187 s | 11.5 GB | ✅ clean |
| 1440×2560 (non-square) | ~4.5 s | ~127 s | ~10 GB | ⚠ patch grid visible |