huluhuluu commited on
Commit
61bcd0f
·
verified ·
1 Parent(s): cddb69d

Add files using upload-large-folder tool

Browse files
Files changed (4) hide show
  1. README.md +83 -0
  2. config.json +33 -0
  3. model.safetensors +3 -0
  4. training_state.pt +3 -0
README.md ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ language:
4
+ - en
5
+ - zh
6
+ library_name: transformers
7
+ pipeline_tag: text-generation
8
+ tags:
9
+ - specforge
10
+ - eagle3
11
+ - speculative-decoding
12
+ - draft-model
13
+ - qwen3
14
+ - sharegpt
15
+ - sliding-window
16
+ base_model:
17
+ - Qwen/Qwen3-1.7B
18
+ ---
19
+
20
+ # Qwen3 1.7B EAGLE3 ShareGPT SW768 - Epoch 9, Step 465000
21
+
22
+ This repository contains a SpecForge EAGLE3 draft-model checkpoint for use with `Qwen/Qwen3-1.7B`.
23
+ It is a draft model for speculative decoding, not a standalone target language model.
24
+
25
+ ## Checkpoint
26
+
27
+ | Field | Value |
28
+ |---|---|
29
+ | Source run | `qwen3-1.7b-eagle3-sharegpt-sw768` |
30
+ | Checkpoint | `epoch_9_step_465000` |
31
+ | Epoch | `9` |
32
+ | Global step | `465000` |
33
+ | Files | `config.json`, `model.safetensors`, `training_state.pt` |
34
+
35
+ ## Training Parameters
36
+
37
+ | Parameter | Value |
38
+ |---|---|
39
+ | Base model | `Qwen/Qwen3-1.7B` |
40
+ | Method | `SpecForge EAGLE3 online training` |
41
+ | Framework revision | `9fbbde8ab5d6ee69fb0af3701330027b8beca37a` |
42
+ | Training data | `sharegpt_train.jsonl` |
43
+ | Learning rate | `0.0001` |
44
+ | Batch size | `1` |
45
+ | Epochs configured | `10` |
46
+ | Total scheduled steps | `467800` |
47
+ | Max length | `2048` |
48
+ | Warmup ratio | `0.015` |
49
+ | Max grad norm | `0.5` |
50
+ | TTT length | `7` |
51
+ | Draft accumulation steps | `1` |
52
+ | Draft sliding window | `768` |
53
+ | Save / eval interval | `5000 / 5000` |
54
+ | Seed | `0` |
55
+ | TP / DP size | `1 / 2` |
56
+ | Attention backend | `sdpa` |
57
+ | Target model backend | `sglang` |
58
+ | SGLang attention backend | `flashinfer` |
59
+ | Dataset build workers | `16` |
60
+
61
+ ## Draft Model Configuration
62
+
63
+ | Field | Value |
64
+ |---|---|
65
+ | Architecture | `LlamaForCausalLMEagle3` |
66
+ | dtype | `bfloat16` |
67
+ | Hidden size | `2048` |
68
+ | Intermediate size | `6144` |
69
+ | Draft layers | `1` |
70
+ | Attention heads | `16` |
71
+ | KV heads | `8` |
72
+ | Draft vocab size | `32000` |
73
+ | Vocab size | `151936` |
74
+ | Max position embeddings | `40960` |
75
+ | Sliding window | `768` |
76
+ | Max window layers | `28` |
77
+
78
+ ## Notes
79
+
80
+ - The checkpoint weights exclude the frozen target embedding weights; SpecForge loads them from the target model.
81
+ - The checkpoint is intended to be loaded by SpecForge/EAGLE3-compatible code.
82
+ - `training_state.pt` is included for provenance and training-state inspection.
83
+ - No benchmark claim is made in this card.
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "LlamaForCausalLMEagle3"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 151643,
8
+ "draft_vocab_size": 32000,
9
+ "dtype": "bfloat16",
10
+ "eos_token_id": 151645,
11
+ "head_dim": 128,
12
+ "hidden_act": "silu",
13
+ "hidden_size": 2048,
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 6144,
16
+ "max_position_embeddings": 40960,
17
+ "max_window_layers": 28,
18
+ "mlp_bias": false,
19
+ "model_type": "llama",
20
+ "num_attention_heads": 16,
21
+ "num_hidden_layers": 1,
22
+ "num_key_value_heads": 8,
23
+ "pretraining_tp": 1,
24
+ "rms_norm_eps": 1e-06,
25
+ "rope_scaling": null,
26
+ "rope_theta": 1000000,
27
+ "sliding_window": 768,
28
+ "tie_word_embeddings": false,
29
+ "transformers_version": "4.57.1",
30
+ "use_cache": true,
31
+ "use_sliding_window": true,
32
+ "vocab_size": 151936
33
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:762e96d7519ac59b611075957cb5fba8441c98d25b172f92da4d6f514b272b3e
3
+ size 274104160
training_state.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e2f5b3d68fdc0b751b1561cd81817b66a50a1c73e9c33cf8bb49a88b5a1e8291
3
+ size 4375