ThePabli commited on
Commit
14a0189
·
verified ·
1 Parent(s): 2b73f43

Add model card

Browse files
Files changed (1) hide show
  1. README.md +41 -0
README.md ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: ATH-MaaS/Marco-Mini-Instruct
4
+ library_name: transformers
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - moe
8
+ - mixture-of-experts
9
+ - reap
10
+ - expert-pruning
11
+ ---
12
+
13
+ # Marco-Mini-Instruct-REAP10
14
+
15
+ [Marco-Mini-Instruct](https://huggingface.co/ATH-MaaS/Marco-Mini-Instruct) (17.3B total, ~0.86B active, 256 experts, 8 active per token) with **10% of its experts removed** by [REAP](https://arxiv.org/abs/2510.13999): 231 of 256 experts kept. These are the full-precision (bf16) safetensors weights, for fine-tuning, re-quantising or running with `transformers`.
16
+
17
+ For ready-to-run quantised files at every pruning ratio, see [kueizen/Marco-Mini-Instruct-REAP-GGUF](https://huggingface.co/kueizen/Marco-Mini-Instruct-REAP-GGUF).
18
+
19
+ ## Load it
20
+
21
+ ```python
22
+ from transformers import AutoModelForCausalLM, AutoTokenizer
23
+
24
+ repo = "kueizen/Marco-Mini-Instruct-REAP10"
25
+ tok = AutoTokenizer.from_pretrained(repo)
26
+ model = AutoModelForCausalLM.from_pretrained(repo, torch_dtype="auto", device_map="auto")
27
+ ```
28
+
29
+ ## How this was made
30
+
31
+ - **Pruning:** [REAP](https://arxiv.org/abs/2510.13999) (Router-weighted Expert Activation Pruning, Cerebras Research), using the [reference implementation](https://github.com/CerebrasResearch/reap). REAP scores each expert by its router weight times the size of its output over a calibration set, then removes the lowest-scoring experts whole. It is one-shot: no retraining. Seed 42.
32
+ - **Calibration set:** [theblackcat102/evol-codealpaca-v1](https://huggingface.co/datasets/theblackcat102/evol-codealpaca-v1) (train split, shuffled, seed 42), 64 samples per category, batch size 1, max sequence length 2048 tokens.
33
+ - **Experts removed:** 10% (231 of 256 kept).
34
+
35
+ We have not run task benchmarks on these checkpoints (MMLU, coding, multilingual). Test them on your own workload before relying on them.
36
+
37
+ ## License and credits
38
+
39
+ Derived from [ATH-MaaS/Marco-Mini-Instruct](https://huggingface.co/ATH-MaaS/Marco-Mini-Instruct) and released under the same **Apache 2.0** licence. What we changed: removed experts with REAP. Nothing else was modified or retrained. REAP is by Cerebras Research: [paper](https://arxiv.org/abs/2510.13999), [code](https://github.com/CerebrasResearch/reap). All credit for the base model goes to its authors.
40
+
41
+ Published by [Kueizen](https://kueizen.com).