huckiyang commited on
Commit
1f2ad6f
·
verified ·
1 Parent(s): 48d11ff

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -10,7 +10,7 @@ pipeline_tag: image-text-to-text
10
 
11
  An **MLX 4-bit** build of Thinking Machines' **Inkling** (975B-total / 41B-active MoE),
12
  quantized from the BF16 checkpoint for **Apple Silicon**. **Omni**: keeps the text decoder
13
- **plus the vision (HMLP) and audio (dMel) towers**. **Self-contained** it bundles the loader
14
  package (`inkling_mlx/`), so it runs with just `mlx` + `mlx-lm` + `transformers`.
15
 
16
  The higher-fidelity sibling of [`mlx-community/Inkling-mlx-2bit`](https://huggingface.co/mlx-community/Inkling-mlx-2bit):
@@ -19,7 +19,7 @@ The higher-fidelity sibling of [`mlx-community/Inkling-mlx-2bit`](https://huggin
19
  ## Quantization (recipe: `experts_only`)
20
 
21
  - **4-bit** affine group quantization (group size 64) on the **routed experts** and the
22
- **vision / audio matmuls** the bulk of the weights.
23
  - **Kept in bf16 (protected):** attention, token / output embeddings, RMSNorms, the router
24
  gate, the per-layer short-convolutions, and the relative-position bias.
25
 
@@ -40,7 +40,7 @@ print(tok.decode(greedy_generate(model, config, ids, max_new_tokens=64)))
40
 
41
  ## Running on a single Mac with SSD expert-offload
42
 
43
- At ~548 GB this doesn't fit resident on one Mac but an MoE only fires **6 of 256** experts
44
  per token. So keep the always-needed weights in RAM (attention, shared experts, embeddings,
45
  norms, router, vision / audio towers) and **page the routed experts from SSD on demand**,
46
  letting the OS page cache hold the hot ones. This runs the omni build on a single Mac Studio.
 
10
 
11
  An **MLX 4-bit** build of Thinking Machines' **Inkling** (975B-total / 41B-active MoE),
12
  quantized from the BF16 checkpoint for **Apple Silicon**. **Omni**: keeps the text decoder
13
+ **plus the vision (HMLP) and audio (dMel) towers**. **Self-contained**. This bundles the loader
14
  package (`inkling_mlx/`), so it runs with just `mlx` + `mlx-lm` + `transformers`.
15
 
16
  The higher-fidelity sibling of [`mlx-community/Inkling-mlx-2bit`](https://huggingface.co/mlx-community/Inkling-mlx-2bit):
 
19
  ## Quantization (recipe: `experts_only`)
20
 
21
  - **4-bit** affine group quantization (group size 64) on the **routed experts** and the
22
+ **vision / audio matmuls** this is the bulk of the weights.
23
  - **Kept in bf16 (protected):** attention, token / output embeddings, RMSNorms, the router
24
  gate, the per-layer short-convolutions, and the relative-position bias.
25
 
 
40
 
41
  ## Running on a single Mac with SSD expert-offload
42
 
43
+ At ~548 GB this doesn't fit resident on one Mac but an MoE only fires **6 of 256** experts
44
  per token. So keep the always-needed weights in RAM (attention, shared experts, embeddings,
45
  norms, router, vision / audio towers) and **page the routed experts from SSD on demand**,
46
  letting the OS page cache hold the hot ones. This runs the omni build on a single Mac Studio.