ycchen commited on
Commit
5fdcaed
·
verified ·
1 Parent(s): 37c0b61

Add files using upload-large-folder tool

Browse files
dflash-32b-draft-v2test-phaseL/config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "DFlashDraftModel"
4
+ ],
5
+ "model_type": "qwen3",
6
+ "hidden_size": 5120,
7
+ "intermediate_size": 13824,
8
+ "num_hidden_layers": 8,
9
+ "num_attention_heads": 40,
10
+ "num_key_value_heads": 8,
11
+ "head_dim": 128,
12
+ "hidden_act": "silu",
13
+ "rms_norm_eps": 1e-06,
14
+ "attention_bias": false,
15
+ "vocab_size": 129280,
16
+ "max_position_embeddings": 65536,
17
+ "rope_theta": 500000.0,
18
+ "rope_scaling": null,
19
+ "sliding_window": 512,
20
+ "layer_types": [
21
+ "sliding_attention",
22
+ "sliding_attention",
23
+ "sliding_attention",
24
+ "sliding_attention",
25
+ "sliding_attention",
26
+ "sliding_attention",
27
+ "sliding_attention",
28
+ "sliding_attention"
29
+ ],
30
+ "block_size": 11,
31
+ "num_target_layers": 64,
32
+ "target_hidden_size": 5120,
33
+ "torch_dtype": "bfloat16",
34
+ "tie_word_embeddings": false,
35
+ "dflash_config": {
36
+ "mask_token_id": 128000,
37
+ "target_layer_ids": [
38
+ 1,
39
+ 10,
40
+ 18,
41
+ 27,
42
+ 35,
43
+ 44,
44
+ 52,
45
+ 61
46
+ ],
47
+ "num_target_layers": 64,
48
+ "block_size": 11,
49
+ "use_attention_sink": true,
50
+ "sliding_window": 512
51
+ }
52
+ }
dflash-32b-draft-v2test-phaseL/model-00001-of-00005.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a549c86d1075338b8f84cba9a02b5ce03d5b408a3d098677cf60eac5acc0dea7
3
+ size 969977384
dflash-32b-draft-v2test-phaseL/model-00002-of-00005.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15271fd91c40f2b71a9274159ce50941798dbec177516bea4aa56c2c8257741f
3
+ size 985718896
dflash-32b-draft-v2test-phaseL/model-00003-of-00005.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e72823f72482b9ab3b17660471bbfcd7f473c175443564be67735f957be0f543
3
+ size 949006504
dflash-32b-draft-v2test-phaseL/model-00004-of-00005.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d59067a7994bb229357b532c9e55de875f7445d89107a3724b58c41af17fa992
3
+ size 959515232
dflash-32b-draft-v2test-phaseL/model-00005-of-00005.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce36865fcd1f6d0a69a736b8d62cd9c7f04c39f214020611906faa0afcc7f17e
3
+ size 959535872
dflash-32b-draft-v2test-phaseL/model.safetensors.index.json ADDED
@@ -0,0 +1,107 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 4823743104
4
+ },
5
+ "weight_map": {
6
+ "fc.weight": "model-00001-of-00005.safetensors",
7
+ "hidden_norm.weight": "model-00001-of-00005.safetensors",
8
+ "layers.0.mlp.down_proj.weight": "model-00001-of-00005.safetensors",
9
+ "layers.0.mlp.gate_proj.weight": "model-00001-of-00005.safetensors",
10
+ "layers.0.mlp.up_proj.weight": "model-00001-of-00005.safetensors",
11
+ "layers.0.post_attention_layernorm.weight": "model-00001-of-00005.safetensors",
12
+ "layers.0.post_feedforward_layernorm.weight": "model-00001-of-00005.safetensors",
13
+ "layers.0.self_attn.k_norm.weight": "model-00001-of-00005.safetensors",
14
+ "layers.0.self_attn.k_proj.weight": "model-00001-of-00005.safetensors",
15
+ "layers.0.self_attn.o_proj.weight": "model-00001-of-00005.safetensors",
16
+ "layers.0.self_attn.q_norm.weight": "model-00001-of-00005.safetensors",
17
+ "layers.0.self_attn.q_proj.weight": "model-00001-of-00005.safetensors",
18
+ "layers.0.self_attn.sinks": "model-00001-of-00005.safetensors",
19
+ "layers.0.self_attn.v_proj.weight": "model-00001-of-00005.safetensors",
20
+ "layers.1.mlp.down_proj.weight": "model-00002-of-00005.safetensors",
21
+ "layers.1.mlp.gate_proj.weight": "model-00002-of-00005.safetensors",
22
+ "layers.1.mlp.up_proj.weight": "model-00002-of-00005.safetensors",
23
+ "layers.1.post_attention_layernorm.weight": "model-00002-of-00005.safetensors",
24
+ "layers.1.post_feedforward_layernorm.weight": "model-00002-of-00005.safetensors",
25
+ "layers.1.self_attn.k_norm.weight": "model-00002-of-00005.safetensors",
26
+ "layers.1.self_attn.k_proj.weight": "model-00002-of-00005.safetensors",
27
+ "layers.1.self_attn.o_proj.weight": "model-00002-of-00005.safetensors",
28
+ "layers.1.self_attn.q_norm.weight": "model-00002-of-00005.safetensors",
29
+ "layers.1.self_attn.q_proj.weight": "model-00002-of-00005.safetensors",
30
+ "layers.1.self_attn.sinks": "model-00002-of-00005.safetensors",
31
+ "layers.1.self_attn.v_proj.weight": "model-00002-of-00005.safetensors",
32
+ "layers.2.mlp.down_proj.weight": "model-00002-of-00005.safetensors",
33
+ "layers.2.mlp.gate_proj.weight": "model-00002-of-00005.safetensors",
34
+ "layers.2.mlp.up_proj.weight": "model-00002-of-00005.safetensors",
35
+ "layers.2.post_attention_layernorm.weight": "model-00002-of-00005.safetensors",
36
+ "layers.2.post_feedforward_layernorm.weight": "model-00002-of-00005.safetensors",
37
+ "layers.2.self_attn.k_norm.weight": "model-00002-of-00005.safetensors",
38
+ "layers.2.self_attn.k_proj.weight": "model-00002-of-00005.safetensors",
39
+ "layers.2.self_attn.o_proj.weight": "model-00003-of-00005.safetensors",
40
+ "layers.2.self_attn.q_norm.weight": "model-00003-of-00005.safetensors",
41
+ "layers.2.self_attn.q_proj.weight": "model-00003-of-00005.safetensors",
42
+ "layers.2.self_attn.sinks": "model-00003-of-00005.safetensors",
43
+ "layers.2.self_attn.v_proj.weight": "model-00003-of-00005.safetensors",
44
+ "layers.3.mlp.down_proj.weight": "model-00003-of-00005.safetensors",
45
+ "layers.3.mlp.gate_proj.weight": "model-00003-of-00005.safetensors",
46
+ "layers.3.mlp.up_proj.weight": "model-00003-of-00005.safetensors",
47
+ "layers.3.post_attention_layernorm.weight": "model-00003-of-00005.safetensors",
48
+ "layers.3.post_feedforward_layernorm.weight": "model-00003-of-00005.safetensors",
49
+ "layers.3.self_attn.k_norm.weight": "model-00003-of-00005.safetensors",
50
+ "layers.3.self_attn.k_proj.weight": "model-00003-of-00005.safetensors",
51
+ "layers.3.self_attn.o_proj.weight": "model-00003-of-00005.safetensors",
52
+ "layers.3.self_attn.q_norm.weight": "model-00003-of-00005.safetensors",
53
+ "layers.3.self_attn.q_proj.weight": "model-00003-of-00005.safetensors",
54
+ "layers.3.self_attn.sinks": "model-00003-of-00005.safetensors",
55
+ "layers.3.self_attn.v_proj.weight": "model-00003-of-00005.safetensors",
56
+ "layers.4.mlp.down_proj.weight": "model-00003-of-00005.safetensors",
57
+ "layers.4.mlp.gate_proj.weight": "model-00003-of-00005.safetensors",
58
+ "layers.4.mlp.up_proj.weight": "model-00004-of-00005.safetensors",
59
+ "layers.4.post_attention_layernorm.weight": "model-00004-of-00005.safetensors",
60
+ "layers.4.post_feedforward_layernorm.weight": "model-00004-of-00005.safetensors",
61
+ "layers.4.self_attn.k_norm.weight": "model-00004-of-00005.safetensors",
62
+ "layers.4.self_attn.k_proj.weight": "model-00004-of-00005.safetensors",
63
+ "layers.4.self_attn.o_proj.weight": "model-00004-of-00005.safetensors",
64
+ "layers.4.self_attn.q_norm.weight": "model-00004-of-00005.safetensors",
65
+ "layers.4.self_attn.q_proj.weight": "model-00004-of-00005.safetensors",
66
+ "layers.4.self_attn.sinks": "model-00004-of-00005.safetensors",
67
+ "layers.4.self_attn.v_proj.weight": "model-00004-of-00005.safetensors",
68
+ "layers.5.mlp.down_proj.weight": "model-00004-of-00005.safetensors",
69
+ "layers.5.mlp.gate_proj.weight": "model-00004-of-00005.safetensors",
70
+ "layers.5.mlp.up_proj.weight": "model-00004-of-00005.safetensors",
71
+ "layers.5.post_attention_layernorm.weight": "model-00004-of-00005.safetensors",
72
+ "layers.5.post_feedforward_layernorm.weight": "model-00004-of-00005.safetensors",
73
+ "layers.5.self_attn.k_norm.weight": "model-00004-of-00005.safetensors",
74
+ "layers.5.self_attn.k_proj.weight": "model-00004-of-00005.safetensors",
75
+ "layers.5.self_attn.o_proj.weight": "model-00004-of-00005.safetensors",
76
+ "layers.5.self_attn.q_norm.weight": "model-00004-of-00005.safetensors",
77
+ "layers.5.self_attn.q_proj.weight": "model-00004-of-00005.safetensors",
78
+ "layers.5.self_attn.sinks": "model-00004-of-00005.safetensors",
79
+ "layers.5.self_attn.v_proj.weight": "model-00004-of-00005.safetensors",
80
+ "layers.6.mlp.down_proj.weight": "model-00004-of-00005.safetensors",
81
+ "layers.6.mlp.gate_proj.weight": "model-00005-of-00005.safetensors",
82
+ "layers.6.mlp.up_proj.weight": "model-00005-of-00005.safetensors",
83
+ "layers.6.post_attention_layernorm.weight": "model-00005-of-00005.safetensors",
84
+ "layers.6.post_feedforward_layernorm.weight": "model-00005-of-00005.safetensors",
85
+ "layers.6.self_attn.k_norm.weight": "model-00005-of-00005.safetensors",
86
+ "layers.6.self_attn.k_proj.weight": "model-00005-of-00005.safetensors",
87
+ "layers.6.self_attn.o_proj.weight": "model-00005-of-00005.safetensors",
88
+ "layers.6.self_attn.q_norm.weight": "model-00005-of-00005.safetensors",
89
+ "layers.6.self_attn.q_proj.weight": "model-00005-of-00005.safetensors",
90
+ "layers.6.self_attn.sinks": "model-00005-of-00005.safetensors",
91
+ "layers.6.self_attn.v_proj.weight": "model-00005-of-00005.safetensors",
92
+ "layers.7.mlp.down_proj.weight": "model-00005-of-00005.safetensors",
93
+ "layers.7.mlp.gate_proj.weight": "model-00005-of-00005.safetensors",
94
+ "layers.7.mlp.up_proj.weight": "model-00005-of-00005.safetensors",
95
+ "layers.7.post_attention_layernorm.weight": "model-00005-of-00005.safetensors",
96
+ "layers.7.post_feedforward_layernorm.weight": "model-00005-of-00005.safetensors",
97
+ "layers.7.self_attn.k_norm.weight": "model-00005-of-00005.safetensors",
98
+ "layers.7.self_attn.k_proj.weight": "model-00005-of-00005.safetensors",
99
+ "layers.7.self_attn.o_proj.weight": "model-00005-of-00005.safetensors",
100
+ "layers.7.self_attn.q_norm.weight": "model-00005-of-00005.safetensors",
101
+ "layers.7.self_attn.q_proj.weight": "model-00005-of-00005.safetensors",
102
+ "layers.7.self_attn.sinks": "model-00005-of-00005.safetensors",
103
+ "layers.7.self_attn.v_proj.weight": "model-00005-of-00005.safetensors",
104
+ "mask_embed": "model-00005-of-00005.safetensors",
105
+ "norm.weight": "model-00005-of-00005.safetensors"
106
+ }
107
+ }