NotaMG commited on
Commit
a085d9d
·
verified ·
1 Parent(s): f42eb97

Add q004 EAGLE3 eagle3-draft-chat-think weights

Browse files
eagle3-draft-chat-think/config.json ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "LlamaForCausalLM"
4
+ ],
5
+ "bos_token_id": 151643,
6
+ "draft_vocab_size": 32768,
7
+ "eagle_config": {
8
+ "base_model": "/home/project-a/Efficient-Qwen-Competition/models/qwen35-4b-awq-text-only-packed",
9
+ "stateless_fusion": true,
10
+ "use_aux_hidden_state": false
11
+ },
12
+ "eos_token_id": 248044,
13
+ "head_dim": 256,
14
+ "hidden_act": "silu",
15
+ "hidden_size": 2560,
16
+ "intermediate_size": 9216,
17
+ "max_position_embeddings": 262144,
18
+ "model_type": "llama",
19
+ "num_attention_heads": 10,
20
+ "num_hidden_layers": 0,
21
+ "num_key_value_heads": 2,
22
+ "pad_token_id": 248044,
23
+ "rms_norm_eps": 1e-06,
24
+ "rope_theta": 1000000.0,
25
+ "tie_word_embeddings": true,
26
+ "torch_dtype": "bfloat16",
27
+ "training": {
28
+ "absolute_positions": true,
29
+ "adam_beta1": 0.9,
30
+ "adam_beta2": 0.95,
31
+ "batch_size": 1,
32
+ "cache_aware_draft_steps": true,
33
+ "dense_target_hidden_soft_loss": true,
34
+ "dense_target_temperature": 0.65,
35
+ "draft_head_dim": 0,
36
+ "draft_num_attention_heads": 0,
37
+ "draft_num_hidden_layers": 0,
38
+ "draft_num_key_value_heads": 0,
39
+ "draft_rope_theta": 0.0,
40
+ "draft_step_decay": 0.9,
41
+ "draft_steps": 4,
42
+ "draft_vocab_jsonl": "/home/project-a/efficient-qwen/tmp/repro_awq_think_4k_20260520/features/chat_think/metadata.jsonl",
43
+ "draft_vocab_size": 32768,
44
+ "drop_longer_than_max_seq_len": false,
45
+ "feature_glob": "/home/project-a/efficient-qwen/tmp/repro_awq_think_4k_20260520/features/chat_think/features_*.pt",
46
+ "gradient_clipping": 0.5,
47
+ "hard_target_weight": 0.75,
48
+ "label_type": "onpolicy_teacher_dense_draft_vocab",
49
+ "learning_rate": 6e-06,
50
+ "max_seq_len": 4096,
51
+ "nemo_draft_loss": false,
52
+ "no_aux_hidden_state": true,
53
+ "num_examples": 4096,
54
+ "official_soft_loss": false,
55
+ "parallel_drafting_anchors_per_example": 1,
56
+ "parallel_drafting_mask_hidden": false,
57
+ "parallel_drafting_objective": false,
58
+ "parallel_drafting_ptd_token_id": -1,
59
+ "save_every": 3000,
60
+ "shift_positions_per_draft_step": false,
61
+ "soft_target_weight": 1.0,
62
+ "steps": 12000,
63
+ "target_hidden_sidecar_map": "",
64
+ "warmup_steps": 500
65
+ },
66
+ "use_cache": true,
67
+ "vocab_size": 248320
68
+ }
eagle3-draft-chat-think/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0aceb4f06084e9b72cb908885c8b12e91cfb508ccc3e6baf4a9c22c8fd21a1bf
3
+ size 194264600