xocialize/Mage-Flow-Edit-mlx

Mage-Flow-Edit — MLX-Swift

Apple-Silicon MLX-Swift support for microsoft/Mage-Flow-Edit (edit, defaults: 20 steps / cfg 5.0 (RL-aligned)). One of six checkpoints of the Mage-Flow family, all sharing one architecture — this repo carries the port's support artifact; the component weights come from the upstream microsoft repo (MIT) and are not re-hosted (multi-component diffusion pipeline → xocialize namespace per mlx-community conventions).

Code, parity gates, and full porting notes: github.com/xocialize/mage-flow-swift. The DiT/VAE/Gaussian-Shading watermark/scheduler are parity-locked against the PyTorch oracle; T2I and CFG paths are gated per-step and validated with renders at the real defaults (sample_512.png is this checkpoint's).

Use

hf download microsoft/Mage-Flow-Edit --local-dir Mage-Flow-Edit
hf download microsoft/Mage-Flow-Edit-Turbo --include "text_encoder/*" --include "vae/*" --local-dir Mage-Flow-Edit   # shared components (byte-identical family-wide)
hf download xocialize/Mage-Flow-Edit-mlx folded_adaln.safetensors --local-dir Mage-Flow-Edit
swift run -c release mage-flow-edit --repo Mage-Flow-Edit --ref dog.jpg --prompt "make the background a snowy forest" --out out.png --steps 20 --cfg 5.0

Notes: the mandatory upstream RAI content filter is wired in (fail-closed); the Gaussian-Shading provenance watermark is reproduced bit-exactly; the DiT runs bf16 with one GEMM row-chunked around mlx-swift's NAX split-K JIT bug (mlx#3797) until a release ships #3810.