---
library_name: mlx
license: mit
pipeline_tag: text-to-image
base_model: microsoft/Mage-Flow-Base
tags:
  - mlx
  - mlx-swift
  - diffusion
  - rectified-flow
  - mage-flow
---

# Mage-Flow-Base — MLX-Swift

Apple-Silicon MLX-Swift support for
[microsoft/Mage-Flow-Base](https://huggingface.co/microsoft/Mage-Flow-Base)
(t2i, defaults: 30 steps / cfg 5.0). 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](https://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

```bash
hf download microsoft/Mage-Flow-Base --local-dir Mage-Flow-Base
hf download microsoft/Mage-Flow-Edit-Turbo --include "text_encoder/*" --include "vae/*" --local-dir Mage-Flow-Base   # shared components (byte-identical family-wide)
hf download xocialize/Mage-Flow-Base-mlx folded_adaln.safetensors --local-dir Mage-Flow-Base
swift run -c release mage-flow-edit --repo Mage-Flow-Base --t2i --prompt "a red fox sitting in a snowy forest, photorealistic" --out out.png --steps 30 --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](https://github.com/ml-explore/mlx/issues/3797)) until a release
ships [#3810](https://github.com/ml-explore/mlx/pull/3810).