jianchen0311 commited on
Commit
0ce151d
·
verified ·
1 Parent(s): 42d3b34

Update weights and config from Modal retrain

Browse files
Files changed (2) hide show
  1. config.json +26 -29
  2. model.safetensors +2 -2
config.json CHANGED
@@ -7,56 +7,53 @@
7
  "auto_map": {
8
  "AutoModel": "dflash.DFlashDraftModel"
9
  },
10
- "block_size": 16,
11
  "dflash_config": {
12
- "mask_token_id": 248070,
 
13
  "target_layer_ids": [
14
  1,
15
- 10,
16
- 19,
17
- 28,
 
 
 
18
  37
19
  ]
20
  },
21
  "dtype": "bfloat16",
22
- "eos_token_id": 248046,
23
  "head_dim": 128,
24
  "hidden_act": "silu",
25
  "hidden_size": 2048,
26
  "initializer_range": 0.02,
27
  "intermediate_size": 6144,
28
  "layer_types": [
29
- "full_attention",
30
- "full_attention",
31
- "full_attention",
32
- "full_attention",
33
- "full_attention",
34
- "full_attention",
35
- "full_attention",
36
  "full_attention"
37
  ],
38
  "max_position_embeddings": 262144,
39
- "max_window_layers": 8,
40
  "model_type": "qwen3",
41
  "num_attention_heads": 32,
42
- "num_hidden_layers": 8,
43
- "num_key_value_heads": 4,
44
  "num_target_layers": 40,
45
- "pad_token_id": 248044,
46
  "rms_norm_eps": 1e-06,
47
- "rope_scaling": {
48
- "beta_fast": 32.0,
49
- "beta_slow": 1.0,
50
- "factor": 64.0,
51
- "original_max_position_embeddings": 4096,
52
- "rope_type": "yarn",
53
- "type": "yarn"
54
  },
55
- "rope_theta": 10000000,
56
- "sliding_window": null,
57
  "tie_word_embeddings": false,
58
- "transformers_version": "4.57.1",
59
- "use_cache": false,
60
- "use_sliding_window": false,
61
  "vocab_size": 248320
62
  }
 
7
  "auto_map": {
8
  "AutoModel": "dflash.DFlashDraftModel"
9
  },
10
+ "bos_token_id": null,
11
  "dflash_config": {
12
+ "block_size": 16,
13
+ "mask_token_id": 248077,
14
  "target_layer_ids": [
15
  1,
16
+ 6,
17
+ 11,
18
+ 16,
19
+ 22,
20
+ 27,
21
+ 32,
22
  37
23
  ]
24
  },
25
  "dtype": "bfloat16",
26
+ "eos_token_id": 248044,
27
  "head_dim": 128,
28
  "hidden_act": "silu",
29
  "hidden_size": 2048,
30
  "initializer_range": 0.02,
31
  "intermediate_size": 6144,
32
  "layer_types": [
33
+ "sliding_attention",
34
+ "sliding_attention",
35
+ "sliding_attention",
36
+ "sliding_attention",
37
+ "sliding_attention",
 
 
38
  "full_attention"
39
  ],
40
  "max_position_embeddings": 262144,
41
+ "max_window_layers": 6,
42
  "model_type": "qwen3",
43
  "num_attention_heads": 32,
44
+ "num_hidden_layers": 6,
45
+ "num_key_value_heads": 8,
46
  "num_target_layers": 40,
47
+ "pad_token_id": null,
48
  "rms_norm_eps": 1e-06,
49
+ "rope_parameters": {
50
+ "rope_theta": 10000000,
51
+ "rope_type": "default"
 
 
 
 
52
  },
53
+ "sliding_window": 4096,
 
54
  "tie_word_embeddings": false,
55
+ "transformers_version": "5.7.0",
56
+ "use_cache": true,
57
+ "use_sliding_window": true,
58
  "vocab_size": 248320
59
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6db5c712b4f3d924026162ad1aedf7fd1fef32437690451137f967d9b7160144
3
- size 948000184
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1fb90ef50a32bfb8dd2abfe601dd3608d6d5b59dc342820a98830f76f8cd72b7
3
+ size 771819674