Faisal commited on
Commit
7d8ed12
·
verified ·
1 Parent(s): 2a368ea

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +108 -0
README.md ADDED
@@ -0,0 +1,108 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ library_name: mlx
4
+ license: other
5
+ license_name: nvidia-open-model-license
6
+ license_link: https://developer.nvidia.com/open-model-license
7
+ pipeline_tag: text-generation
8
+ base_model: nvidia/NVIDIA-Nemotron-3-Super-120B-A12B-BF16
9
+ tags:
10
+ - mlx
11
+ - safetensors
12
+ - nemotron_h
13
+ - nemotron
14
+ - mamba
15
+ - mamba2
16
+ - mixture-of-experts
17
+ - 6bit
18
+ - quantized
19
+ - apple-silicon
20
+ - text-generation
21
+ - conversational
22
+ - reasoning
23
+ - lm-studio
24
+ - custom_code
25
+ ---
26
+
27
+ # Nemotron-3-Super-120B-A12B — MLX 6-bit
28
+
29
+ MLX quantization of [nvidia/NVIDIA-Nemotron-3-Super-120B-A12B-BF16](https://huggingface.co/nvidia/NVIDIA-Nemotron-3-Super-120B-A12B-BF16) for Apple Silicon.
30
+
31
+ ## Key Specs
32
+
33
+ | Detail | Value |
34
+ |---|---|
35
+ | Architecture | Hybrid Mamba-2 + Transformer Attention + Latent MoE |
36
+ | Total Parameters | 120B |
37
+ | Active Parameters | 12B per token |
38
+ | Context Length | 1M tokens (262,144 default) |
39
+ | Experts | 512 routed, 22 active per token, 1 shared |
40
+ | Quantization | 6-bit affine (6.507 BPW), group size 64 |
41
+ | Disk Size | ~92 GB |
42
+ | Peak Memory | ~98.4 GB |
43
+
44
+ ## Requirements
45
+
46
+ - Apple Silicon Mac with **128GB+ unified memory**
47
+ - `mlx-lm >= 0.31.2` (install from git main for Latent MoE support)
48
+
49
+ ```bash
50
+ pip install git+https://github.com/ml-explore/mlx-lm.git
51
+ ```
52
+
53
+ ## Usage
54
+
55
+ ### CLI
56
+
57
+ ```bash
58
+ mlx_lm.generate \
59
+ --model FF-01/Nemotron-3-Super-120B-A12B-MLX-6bit \
60
+ --prompt "Hello!" \
61
+ --max-tokens 256
62
+ ```
63
+
64
+ ### Python
65
+
66
+ ```python
67
+ from mlx_lm import load, generate
68
+
69
+ model, tokenizer = load("FF-01/Nemotron-3-Super-120B-A12B-MLX-6bit")
70
+ response = generate(model, tokenizer, prompt="Hello!", max_tokens=256)
71
+ print(response)
72
+ ```
73
+
74
+ ### LM Studio
75
+
76
+ This model is compatible with [LM Studio](https://lmstudio.ai) on Apple Silicon. Search for `FF-01/Nemotron-3-Super-120B-A12B-MLX-6bit` in the model browser and download directly.
77
+
78
+ ## Performance
79
+
80
+ Tested on M5 Pro Max (128GB):
81
+
82
+ | Metric | Value |
83
+ |---|---|
84
+ | Generation Speed | ~43.6 tok/s |
85
+ | Peak Memory | 98.4 GB |
86
+
87
+ ## About the Architecture
88
+
89
+ Nemotron-H is a hybrid architecture combining three components:
90
+
91
+ - **Mamba-2 layers** — efficient state-space model for long-context processing
92
+ - **Transformer attention layers** — standard multi-head attention (GQA, 32 heads, 2 KV heads)
93
+ - **Latent MoE** — 512 experts with latent routing, 22 active per token, plus 1 shared expert
94
+
95
+ The layer pattern alternates between Mamba (M) and attention with MoE (E) blocks across 88 layers. This hybrid design achieves strong performance with only 12B active parameters per token despite having 120B total.
96
+
97
+ ## Reasoning Model
98
+
99
+ This is a reasoning model that outputs chain-of-thought before the final answer. The model uses `<think>` and `</think>` tags to delineate reasoning.
100
+
101
+ ## License
102
+
103
+ [NVIDIA Open Model License](https://developer.nvidia.com/open-model-license)
104
+
105
+ ## Credits
106
+
107
+ - Base model by [NVIDIA](https://huggingface.co/nvidia)
108
+ - MLX quantization by [FF-01](https://huggingface.co/FF-01)