VisualInference commited on
Commit
4e983d0
·
verified ·
1 Parent(s): 6cbdece

Card: whole-process peak memory and --low-memory (pjit-mlx 0.1.1)

Browse files
Files changed (1) hide show
  1. README.md +6 -3
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
  ![Eight renders from this tier, made on an M4 Pro from prompts and seeds in Linum's gallery](assets/gallery.jpg)
@@ -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
  ![Eight renders from this tier, made on an M4 Pro from prompts and seeds in Linum's gallery](assets/gallery.jpg)
 
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