RelaxingSnorlax commited on
Commit
cfb5c81
·
verified ·
1 Parent(s): 4f0f1c9

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +19 -4
config.json CHANGED
@@ -1,9 +1,8 @@
1
  {
2
  "speculators_model_type": "mtp",
3
  "architectures": [
4
- "FastMTPSpeculator"
5
  ],
6
- "checkpoint_format": "tencentbac",
7
  "num_speculative_steps": 3,
8
  "num_nextn_predict_layers": 1,
9
  "mtp_loss_step_weights": [
@@ -19,7 +18,7 @@
19
  "max_position_embeddings": 32768,
20
  "rms_norm_eps": 1e-05,
21
  "transformer_config": {
22
- "model_type": "qwen2",
23
  "vocab_size": 151680,
24
  "hidden_size": 4096,
25
  "intermediate_size": 11008,
@@ -31,5 +30,21 @@
31
  "rope_theta": 640000,
32
  "tie_word_embeddings": false,
33
  "attention_bias": true
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
34
  }
35
- }
 
1
  {
2
  "speculators_model_type": "mtp",
3
  "architectures": [
4
+ "MTPSpeculator"
5
  ],
 
6
  "num_speculative_steps": 3,
7
  "num_nextn_predict_layers": 1,
8
  "mtp_loss_step_weights": [
 
18
  "max_position_embeddings": 32768,
19
  "rms_norm_eps": 1e-05,
20
  "transformer_config": {
21
+ "model_type": "mimo",
22
  "vocab_size": 151680,
23
  "hidden_size": 4096,
24
  "intermediate_size": 11008,
 
30
  "rope_theta": 640000,
31
  "tie_word_embeddings": false,
32
  "attention_bias": true
33
+ },
34
+ "speculators_config": {
35
+ "algorithm": "mtp",
36
+ "proposal_methods": [
37
+ {
38
+ "proposal_type": "greedy",
39
+ "speculative_tokens": 3,
40
+ "verifier_accept_k": 1,
41
+ "accept_tolerance": 0.0
42
+ }
43
+ ],
44
+ "default_proposal_method": "greedy",
45
+ "verifier": {
46
+ "name_or_path": "TencentBAC/FastMTP",
47
+ "architectures": ["MiMoForCausalLM"]
48
+ }
49
  }
50
+ }