0xSero commited on
Commit
41e33ad
·
verified ·
1 Parent(s): 1ff461b

Standardize model card (template rollout)

Browse files
Files changed (1) hide show
  1. README.md +46 -29
README.md CHANGED
@@ -3,28 +3,52 @@ language:
3
  - en
4
  license: gemma
5
  tags:
6
- - safetensors
 
 
7
  - gemma4
8
  - moe
9
  - pruning
10
  - reap
11
- - cerebras
12
- - expert-pruning
13
  base_model:
14
  - google/gemma-4-26b-a4b-it
15
  library_name: transformers
16
  pipeline_tag: text-generation
17
  ---
 
18
  > [!TIP]
19
- > Support this work: **[donate.sybilsolutions.ai](https://donate.sybilsolutions.ai)**
20
- >
21
- > REAP surfaces: [GLM](https://huggingface.co/spaces/0xSero/reap-glm-family) | [MiniMax](https://huggingface.co/spaces/0xSero/reap-minimax-family) | [Qwen](https://huggingface.co/spaces/0xSero/reap-qwen-family) | [Gemma](https://huggingface.co/spaces/0xSero/reap-gemma-family) | [Paper](https://arxiv.org/abs/2510.13999) | [Code](https://github.com/CerebrasResearch/reap) | [PR17](https://github.com/CerebrasResearch/reap/pull/17) | [Cerebras Collection](https://huggingface.co/collections/cerebras/cerebras-reap)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
22
 
23
- # Gemma 4 21B-A4B-it REAP
 
 
 
 
 
 
24
 
25
  **20% expert-pruned** version of [google/gemma-4-26b-a4b-it](https://huggingface.co/google/gemma-4-26b-a4b-it) using **[Cerebras REAP](https://github.com/cerebras/reap)** (Router-weighted Expert Activation Pruning).
26
 
27
- | | Original | This Model (0.20) | [0.30 variant](https://huggingface.co/0xSero/gemma-4-19b-a4b-it-REAP) |
28
  |---|---:|---:|---:|
29
  | **Total params** | ~26B | **21.34B** | 19.02B |
30
  | **Experts per layer** | 128 | **103** | 90 |
@@ -131,7 +155,7 @@ Gemma 4 uses a hybrid sliding/full attention MoE architecture:
131
  ```python
132
  from transformers import AutoModelForCausalLM, AutoTokenizer
133
 
134
- model_id = "0xSero/gemma-4-21b-a4b-it-REAP"
135
  tokenizer = AutoTokenizer.from_pretrained(model_id)
136
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype="auto", device_map="auto", trust_remote_code=True)
137
 
@@ -147,7 +171,7 @@ print(tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_toke
147
  ```bash
148
  pip install vllm>=0.19 transformers>=5.0
149
 
150
- vllm serve 0xSero/gemma-4-21b-a4b-it-REAP \
151
  --tensor-parallel-size 2 \
152
  --enforce-eager \
153
  --gpu-memory-utilization 0.9 \
@@ -155,31 +179,24 @@ vllm serve 0xSero/gemma-4-21b-a4b-it-REAP \
155
  --trust-remote-code
156
  ```
157
 
158
- ## Citation
159
-
160
- ```bibtex
161
- @inproceedings{lasby2025reap,
162
- title={{REAP} the Experts: Why Pruning Prevails for One-Shot {MoE} Compression},
163
- author={Lasby, Mike and others},
164
- booktitle={International Conference on Learning Representations (ICLR)},
165
- year={2026},
166
- url={https://arxiv.org/abs/2510.13999}
167
- }
168
- ```
169
 
170
  ## Links
171
 
172
  - **REAP paper:** [arxiv.org/abs/2510.13999](https://arxiv.org/abs/2510.13999)
173
  - **REAP code:** [github.com/cerebras/reap](https://github.com/cerebras/reap)
174
- - **30% pruned variant:** [0xSero/gemma-4-19b-a4b-it-REAP](https://huggingface.co/0xSero/gemma-4-19b-a4b-it-REAP)
175
  - **Base model:** [google/gemma-4-26b-a4b-it](https://huggingface.co/google/gemma-4-26b-a4b-it)
176
 
177
- ## Sponsors
 
178
 
179
- Thank you for the kind sponsors, wouldn't be possible without them:
 
 
 
 
 
 
180
 
181
- - Nvidia
182
- - TNG Technology
183
- - Lambda
184
- - Prime Intellect
185
- - HotAisle
 
3
  - en
4
  license: gemma
5
  tags:
6
+ - cerebras
7
+ - expert-pruning
8
+ - gemma
9
  - gemma4
10
  - moe
11
  - pruning
12
  - reap
13
+ - safetensors
 
14
  base_model:
15
  - google/gemma-4-26b-a4b-it
16
  library_name: transformers
17
  pipeline_tag: text-generation
18
  ---
19
+
20
  > [!TIP]
21
+ > **[Support this work ](https://donate.sybilsolutions.ai)** · [X](https://x.com/0xsero) · [GitHub](https://github.com/0xsero) · [REAP paper](https://arxiv.org/abs/2510.13999) · [Cerebras REAP](https://huggingface.co/collections/cerebras/cerebras-reap)
22
+
23
+ # Gemma-4-21B
24
+
25
+ REAP-pruned [google/gemma-4-26b-a4b-it](https://huggingface.co/google/gemma-4-26b-a4b-it).
26
+
27
+ ## At a glance
28
+
29
+ | | |
30
+ |---|---|
31
+ | Base model | — |
32
+ | Format | BF16 |
33
+ | Total params | **21B** |
34
+ | Active / token | — |
35
+ | Experts / layer | 103 |
36
+ | Layers | — |
37
+ | Hidden size | — |
38
+ | Context | — |
39
+ | On-disk size | 43 GB |
40
 
41
+ ## Which variant should I pick?
42
+
43
+ | Variant | Format | Link |
44
+ |---|---|---|
45
+ | `Gemma-4-19B` | BF16 | [link](https://huggingface.co/0xSero/Gemma-4-19B) |
46
+ | `Gemma-4-21B` **(this)** | BF16 | [link](https://huggingface.co/0xSero/Gemma-4-21B) |
47
+ | `gemma-moe-reap` | BF16 | [link](https://huggingface.co/0xSero/gemma-moe-reap) |
48
 
49
  **20% expert-pruned** version of [google/gemma-4-26b-a4b-it](https://huggingface.co/google/gemma-4-26b-a4b-it) using **[Cerebras REAP](https://github.com/cerebras/reap)** (Router-weighted Expert Activation Pruning).
50
 
51
+ | | Original | This Model (0.20) | [0.30 variant](https://huggingface.co/0xSero/Gemma-4-19B) |
52
  |---|---:|---:|---:|
53
  | **Total params** | ~26B | **21.34B** | 19.02B |
54
  | **Experts per layer** | 128 | **103** | 90 |
 
155
  ```python
156
  from transformers import AutoModelForCausalLM, AutoTokenizer
157
 
158
+ model_id = "0xSero/Gemma-4-21B"
159
  tokenizer = AutoTokenizer.from_pretrained(model_id)
160
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype="auto", device_map="auto", trust_remote_code=True)
161
 
 
171
  ```bash
172
  pip install vllm>=0.19 transformers>=5.0
173
 
174
+ vllm serve 0xSero/Gemma-4-21B \
175
  --tensor-parallel-size 2 \
176
  --enforce-eager \
177
  --gpu-memory-utilization 0.9 \
 
179
  --trust-remote-code
180
  ```
181
 
 
 
 
 
 
 
 
 
 
 
 
182
 
183
  ## Links
184
 
185
  - **REAP paper:** [arxiv.org/abs/2510.13999](https://arxiv.org/abs/2510.13999)
186
  - **REAP code:** [github.com/cerebras/reap](https://github.com/cerebras/reap)
187
+ - **30% pruned variant:** [0xSero/Gemma-4-19B](https://huggingface.co/0xSero/Gemma-4-19B)
188
  - **Base model:** [google/gemma-4-26b-a4b-it](https://huggingface.co/google/gemma-4-26b-a4b-it)
189
 
190
+ ## License & citation
191
+ License inherited from the base model.
192
 
193
+ ```bibtex
194
+ @misc{lasby2025reap,
195
+ title = {REAP the Experts: Why Pruning Prevails for One-Shot MoE Compression},
196
+ author = {Mike Lasby and Ivan Lazarevich and Nish Sinnadurai and Sean Lie and Yani Ioannou and Vithursan Thangarasa},
197
+ year = {2025}, eprint = {2510.13999}, archivePrefix = {arXiv}
198
+ }
199
+ ```
200
 
201
+ ## Sponsors
202
+ Made possible by **NVIDIA · TNG Technology · Lambda · Prime Intellect · Hot Aisle**.