Instructions to use VisualInference/P-JiT-21K-MLX-8bit with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- MLX
How to use VisualInference/P-JiT-21K-MLX-8bit with MLX:
# Download the model from the Hub pip install huggingface_hub[hf_xet] hf download VisualInference/P-JiT-21K-MLX-8bit --local-dir P-JiT-21K-MLX-8bit
- Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- LM Studio
- Atomic Chat
Card: whole-process peak memory and --low-memory (pjit-mlx 0.1.1)
Browse files
README.md
CHANGED
|
@@ -12,7 +12,7 @@ This is an MLX export of [Linum-AI/pjit-21k](https://huggingface.co/Linum-AI/pji
|
|
| 12 |
text-to-image DiT. It comes with its Qwen3.5-4B caption encoder, trimmed to the 28 blocks the model reads. The runner
|
| 13 |
is [pjit-mlx](https://github.com/felix-ab/pjit-mlx).
|
| 14 |
|
| 15 |
-
**Tier:** 8-bit (compact: 3.6 GB peak). It is as clean as bf16 at 100% and usually gives the same picture for a seed
|
| 16 |
(14 of 15 gallery tests, with more fine-detail drift), but it can fork on sensitive prompts.
|
| 17 |
|
| 18 |

|
|
@@ -22,12 +22,15 @@ is [pjit-mlx](https://github.com/felix-ab/pjit-mlx).
|
|
| 22 |
| M4 Pro, one 1152×864 image, 50 steps | This tier | PyTorch on MPS |
|
| 23 |
|---|---|---|
|
| 24 |
| Total time | 79 s | 139 s |
|
| 25 |
-
| Peak memory | 3.6 GB | 14.9 GB |
|
| 26 |
| Download | 6.9 GB | 18.0 GB |
|
| 27 |
|
|
|
|
|
|
|
|
|
|
| 28 |
```bash
|
| 29 |
pip install git+https://github.com/felix-ab/pjit-mlx
|
| 30 |
-
pjit-mlx --tier 8bit --seeds 4 --aspect 4:3 \
|
| 31 |
--prompt "Bright midday sun lights a white man in his thirties in a head-and-shoulders portrait, squinting slightly with a relaxed half smile. His faded blue chambray shirt is open at the collar, and a field of tall grass blurs behind him."
|
| 32 |
```
|
| 33 |
|
|
|
|
| 12 |
text-to-image DiT. It comes with its Qwen3.5-4B caption encoder, trimmed to the 28 blocks the model reads. The runner
|
| 13 |
is [pjit-mlx](https://github.com/felix-ab/pjit-mlx).
|
| 14 |
|
| 15 |
+
**Tier:** 8-bit (compact: 3.6 GB peak for the whole process with `--low-memory`, 5.0 GB without). It is as clean as bf16 at 100% and usually gives the same picture for a seed
|
| 16 |
(14 of 15 gallery tests, with more fine-detail drift), but it can fork on sensitive prompts.
|
| 17 |
|
| 18 |

|
|
|
|
| 22 |
| M4 Pro, one 1152×864 image, 50 steps | This tier | PyTorch on MPS |
|
| 23 |
|---|---|---|
|
| 24 |
| Total time | 79 s | 139 s |
|
| 25 |
+
| Peak memory, whole process | 5.0 GB; 3.6 GB with `--low-memory` | 14.9 GB |
|
| 26 |
| Download | 6.9 GB | 18.0 GB |
|
| 27 |
|
| 28 |
+
Peak memory is the process's peak physical footprint as macOS reports it, the same meter in both columns.
|
| 29 |
+
`--low-memory` evaluates the DiT block by block and caps MLX's buffer cache; the pixels are identical.
|
| 30 |
+
|
| 31 |
```bash
|
| 32 |
pip install git+https://github.com/felix-ab/pjit-mlx
|
| 33 |
+
pjit-mlx --tier 8bit --low-memory --seeds 4 --aspect 4:3 \
|
| 34 |
--prompt "Bright midday sun lights a white man in his thirties in a head-and-shoulders portrait, squinting slightly with a relaxed half smile. His faded blue chambray shirt is open at the collar, and a field of tall grass blurs behind him."
|
| 35 |
```
|
| 36 |
|