vxbrandon commited on
Commit
4f8396a
1 Parent(s): cb2675b

Training in progress, step 550

Browse files
adapter_config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "mistralai/Mistral-7B-v0.1",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layers_pattern": null,
10
+ "layers_to_transform": null,
11
+ "loftq_config": {},
12
+ "lora_alpha": 16,
13
+ "lora_dropout": 0.1,
14
+ "megatron_config": null,
15
+ "megatron_core": "megatron.core",
16
+ "modules_to_save": null,
17
+ "peft_type": "LORA",
18
+ "r": 64,
19
+ "rank_pattern": {},
20
+ "revision": null,
21
+ "target_modules": [
22
+ "q_proj",
23
+ "up_proj",
24
+ "gate_proj",
25
+ "down_proj",
26
+ "v_proj"
27
+ ],
28
+ "task_type": "CAUSAL_LM"
29
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:202d396a5e1092600aa9a5c2e16776525c65d2365b419b7b3a3d8b2f44ea0343
3
+ size 281061608
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "mistralai/Mistral-7B-v0.1",
3
  "architectures": [
4
  "SparseMistralforCausalLM"
5
  ],
@@ -23,43 +23,44 @@
23
  "rope_theta": 10000.0,
24
  "sliding_window": 4096,
25
  "thresholds": [
26
- 0.027081236243247986,
27
  0.037111327052116394,
28
- 0.07923770695924759,
29
- 0.07923770695924759,
30
- 0.09127381443977356,
31
- 0.10531593859195709,
32
- 0.10932797938585281,
33
- 0.1153460294008255,
34
- 0.11935807019472122,
35
- 0.11935807019472122,
36
- 0.12938815355300903,
37
  0.13340020179748535,
38
- 0.14343027770519257,
39
- 0.14343027770519257,
40
- 0.15346036851406097,
41
- 0.17151454091072083,
42
- 0.19157472252845764,
43
- 0.21564693748950958,
44
- 0.22768303751945496,
 
 
 
 
45
  0.24172517657279968,
46
- 0.24774321913719177,
47
- 0.2577733099460602,
48
- 0.2577733099460602,
49
- 0.26379138231277466,
50
- 0.26379138231277466,
51
- 0.26379138231277466,
52
- 0.26379138231277466,
53
- 0.26379138231277466,
54
  0.26579737663269043,
55
- 0.26980942487716675,
56
- 0.30190572142601013,
57
- 0.5386158227920532
 
 
 
 
 
 
 
58
  ],
59
  "tie_word_embeddings": false,
60
  "torch_dtype": "bfloat16",
61
  "transformers_version": "4.36.2",
62
  "use_cache": false,
 
63
  "use_sparse_model": true,
64
  "use_sparse_predictor": false,
65
  "use_sparse_regularization": false,
 
1
  {
2
+ "_name_or_path": "/scr/jay/ckpt/Mistral_Sparse_refined_web_70p",
3
  "architectures": [
4
  "SparseMistralforCausalLM"
5
  ],
 
23
  "rope_theta": 10000.0,
24
  "sliding_window": 4096,
25
  "thresholds": [
 
26
  0.037111327052116394,
27
+ 0.05315947160124779,
28
+ 0.06720159947872162,
29
+ 0.08525575697422028,
30
+ 0.11133399605751038,
 
 
 
 
 
31
  0.13340020179748535,
32
+ 0.14744232594966888,
33
+ 0.16549648344516754,
34
+ 0.17552657425403595,
35
+ 0.1775325983762741,
36
+ 0.18154463171958923,
37
+ 0.18555666506290436,
38
+ 0.18956869840621948,
39
+ 0.1935807317495346,
40
+ 0.2096288800239563,
41
+ 0.22166499495506287,
42
+ 0.23771312832832336,
43
  0.24172517657279968,
44
+ 0.25376129150390625,
45
+ 0.25977933406829834,
46
+ 0.2617853581905365,
 
 
 
 
 
47
  0.26579737663269043,
48
+ 0.26579737663269043,
49
+ 0.26579737663269043,
50
+ 0.2678034007549286,
51
+ 0.2678034007549286,
52
+ 0.2678034007549286,
53
+ 0.2678034007549286,
54
+ 0.2678034007549286,
55
+ 0.2678034007549286,
56
+ 0.2678034007549286,
57
+ 0.2678034007549286
58
  ],
59
  "tie_word_embeddings": false,
60
  "torch_dtype": "bfloat16",
61
  "transformers_version": "4.36.2",
62
  "use_cache": false,
63
+ "use_relu": false,
64
  "use_sparse_model": true,
65
  "use_sparse_predictor": false,
66
  "use_sparse_regularization": false,
model-00001-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c0c61edb0e98751a0f778d8d0a0b9ea175e91980b755c615051f8b7d1bd57e0b
3
- size 4943163992
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:926ad289ffdf77974545ba4b5536df9e17efd3e57ca8c03c5ffab79565bd4ebe
3
+ size 4943162336
model-00002-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:454e336cba7cbc747b1cc746101181c9c2f9aa960ebe7bcee79bee0633c618b4
3
- size 4999821144
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67d121ed5fdea2c4313698d45947e3fd6fc2acfa32ce70226d98031a67768ad9
3
+ size 4999819336
model-00003-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ae884c75cdae5bf5f5657fd22168350f4c2d3585dfe0f9f1b552d4494c41903
3
- size 4540517840
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:201c5c3752091873297146720180b767f0097ebc97add003b1dd83d2a93e3193
3
+ size 4540516344
model.safetensors.index.json CHANGED
@@ -3,296 +3,296 @@
3
  "total_size": 14483464192
4
  },
5
  "weight_map": {
6
- "base_model.model.lm_head.weight": "model-00003-of-00003.safetensors",
7
- "base_model.model.model.embed_tokens.weight": "model-00001-of-00003.safetensors",
8
- "base_model.model.model.layers.0.input_layernorm.weight": "model-00001-of-00003.safetensors",
9
- "base_model.model.model.layers.0.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
10
- "base_model.model.model.layers.0.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
11
- "base_model.model.model.layers.0.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
12
- "base_model.model.model.layers.0.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
13
- "base_model.model.model.layers.0.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
14
- "base_model.model.model.layers.0.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
15
- "base_model.model.model.layers.0.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
16
- "base_model.model.model.layers.0.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
17
- "base_model.model.model.layers.1.input_layernorm.weight": "model-00001-of-00003.safetensors",
18
- "base_model.model.model.layers.1.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
19
- "base_model.model.model.layers.1.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
20
- "base_model.model.model.layers.1.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
21
- "base_model.model.model.layers.1.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
22
- "base_model.model.model.layers.1.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
23
- "base_model.model.model.layers.1.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
24
- "base_model.model.model.layers.1.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
25
- "base_model.model.model.layers.1.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
26
- "base_model.model.model.layers.10.input_layernorm.weight": "model-00002-of-00003.safetensors",
27
- "base_model.model.model.layers.10.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
28
- "base_model.model.model.layers.10.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
29
- "base_model.model.model.layers.10.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
30
- "base_model.model.model.layers.10.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
31
- "base_model.model.model.layers.10.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
32
- "base_model.model.model.layers.10.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
33
- "base_model.model.model.layers.10.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
34
- "base_model.model.model.layers.10.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
35
- "base_model.model.model.layers.11.input_layernorm.weight": "model-00002-of-00003.safetensors",
36
- "base_model.model.model.layers.11.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
37
- "base_model.model.model.layers.11.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
38
- "base_model.model.model.layers.11.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
39
- "base_model.model.model.layers.11.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
40
- "base_model.model.model.layers.11.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
41
- "base_model.model.model.layers.11.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
42
- "base_model.model.model.layers.11.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
43
- "base_model.model.model.layers.11.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
44
- "base_model.model.model.layers.12.input_layernorm.weight": "model-00002-of-00003.safetensors",
45
- "base_model.model.model.layers.12.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
46
- "base_model.model.model.layers.12.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
47
- "base_model.model.model.layers.12.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
48
- "base_model.model.model.layers.12.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
49
- "base_model.model.model.layers.12.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
50
- "base_model.model.model.layers.12.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
51
- "base_model.model.model.layers.12.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
52
- "base_model.model.model.layers.12.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
53
- "base_model.model.model.layers.13.input_layernorm.weight": "model-00002-of-00003.safetensors",
54
- "base_model.model.model.layers.13.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
55
- "base_model.model.model.layers.13.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
56
- "base_model.model.model.layers.13.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
57
- "base_model.model.model.layers.13.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
58
- "base_model.model.model.layers.13.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
59
- "base_model.model.model.layers.13.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
60
- "base_model.model.model.layers.13.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
61
- "base_model.model.model.layers.13.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
62
- "base_model.model.model.layers.14.input_layernorm.weight": "model-00002-of-00003.safetensors",
63
- "base_model.model.model.layers.14.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
64
- "base_model.model.model.layers.14.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
65
- "base_model.model.model.layers.14.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
66
- "base_model.model.model.layers.14.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
67
- "base_model.model.model.layers.14.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
68
- "base_model.model.model.layers.14.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
69
- "base_model.model.model.layers.14.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
70
- "base_model.model.model.layers.14.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
71
- "base_model.model.model.layers.15.input_layernorm.weight": "model-00002-of-00003.safetensors",
72
- "base_model.model.model.layers.15.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
73
- "base_model.model.model.layers.15.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
74
- "base_model.model.model.layers.15.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
75
- "base_model.model.model.layers.15.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
76
- "base_model.model.model.layers.15.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
77
- "base_model.model.model.layers.15.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
78
- "base_model.model.model.layers.15.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
79
- "base_model.model.model.layers.15.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
80
- "base_model.model.model.layers.16.input_layernorm.weight": "model-00002-of-00003.safetensors",
81
- "base_model.model.model.layers.16.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
82
- "base_model.model.model.layers.16.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
83
- "base_model.model.model.layers.16.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
84
- "base_model.model.model.layers.16.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
85
- "base_model.model.model.layers.16.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
86
- "base_model.model.model.layers.16.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
87
- "base_model.model.model.layers.16.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
88
- "base_model.model.model.layers.16.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
89
- "base_model.model.model.layers.17.input_layernorm.weight": "model-00002-of-00003.safetensors",
90
- "base_model.model.model.layers.17.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
91
- "base_model.model.model.layers.17.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
92
- "base_model.model.model.layers.17.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
93
- "base_model.model.model.layers.17.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
94
- "base_model.model.model.layers.17.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
95
- "base_model.model.model.layers.17.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
96
- "base_model.model.model.layers.17.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
97
- "base_model.model.model.layers.17.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
98
- "base_model.model.model.layers.18.input_layernorm.weight": "model-00002-of-00003.safetensors",
99
- "base_model.model.model.layers.18.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
100
- "base_model.model.model.layers.18.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
101
- "base_model.model.model.layers.18.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
102
- "base_model.model.model.layers.18.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
103
- "base_model.model.model.layers.18.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
104
- "base_model.model.model.layers.18.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
105
- "base_model.model.model.layers.18.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
106
- "base_model.model.model.layers.18.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
107
- "base_model.model.model.layers.19.input_layernorm.weight": "model-00002-of-00003.safetensors",
108
- "base_model.model.model.layers.19.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
109
- "base_model.model.model.layers.19.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
110
- "base_model.model.model.layers.19.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
111
- "base_model.model.model.layers.19.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
112
- "base_model.model.model.layers.19.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
113
- "base_model.model.model.layers.19.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
114
- "base_model.model.model.layers.19.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
115
- "base_model.model.model.layers.19.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
116
- "base_model.model.model.layers.2.input_layernorm.weight": "model-00001-of-00003.safetensors",
117
- "base_model.model.model.layers.2.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
118
- "base_model.model.model.layers.2.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
119
- "base_model.model.model.layers.2.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
120
- "base_model.model.model.layers.2.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
121
- "base_model.model.model.layers.2.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
122
- "base_model.model.model.layers.2.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
123
- "base_model.model.model.layers.2.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
124
- "base_model.model.model.layers.2.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
125
- "base_model.model.model.layers.20.input_layernorm.weight": "model-00002-of-00003.safetensors",
126
- "base_model.model.model.layers.20.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
127
- "base_model.model.model.layers.20.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
128
- "base_model.model.model.layers.20.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
129
- "base_model.model.model.layers.20.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
130
- "base_model.model.model.layers.20.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
131
- "base_model.model.model.layers.20.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
132
- "base_model.model.model.layers.20.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
133
- "base_model.model.model.layers.20.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
134
- "base_model.model.model.layers.21.input_layernorm.weight": "model-00002-of-00003.safetensors",
135
- "base_model.model.model.layers.21.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
136
- "base_model.model.model.layers.21.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
137
- "base_model.model.model.layers.21.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
138
- "base_model.model.model.layers.21.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
139
- "base_model.model.model.layers.21.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
140
- "base_model.model.model.layers.21.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
141
- "base_model.model.model.layers.21.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
142
- "base_model.model.model.layers.21.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
143
- "base_model.model.model.layers.22.input_layernorm.weight": "model-00003-of-00003.safetensors",
144
- "base_model.model.model.layers.22.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
145
- "base_model.model.model.layers.22.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
146
- "base_model.model.model.layers.22.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
147
- "base_model.model.model.layers.22.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
148
- "base_model.model.model.layers.22.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
149
- "base_model.model.model.layers.22.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
150
- "base_model.model.model.layers.22.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
151
- "base_model.model.model.layers.22.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
152
- "base_model.model.model.layers.23.input_layernorm.weight": "model-00003-of-00003.safetensors",
153
- "base_model.model.model.layers.23.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
154
- "base_model.model.model.layers.23.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
155
- "base_model.model.model.layers.23.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
156
- "base_model.model.model.layers.23.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
157
- "base_model.model.model.layers.23.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
158
- "base_model.model.model.layers.23.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
159
- "base_model.model.model.layers.23.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
160
- "base_model.model.model.layers.23.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
161
- "base_model.model.model.layers.24.input_layernorm.weight": "model-00003-of-00003.safetensors",
162
- "base_model.model.model.layers.24.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
163
- "base_model.model.model.layers.24.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
164
- "base_model.model.model.layers.24.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
165
- "base_model.model.model.layers.24.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
166
- "base_model.model.model.layers.24.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
167
- "base_model.model.model.layers.24.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
168
- "base_model.model.model.layers.24.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
169
- "base_model.model.model.layers.24.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
170
- "base_model.model.model.layers.25.input_layernorm.weight": "model-00003-of-00003.safetensors",
171
- "base_model.model.model.layers.25.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
172
- "base_model.model.model.layers.25.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
173
- "base_model.model.model.layers.25.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
174
- "base_model.model.model.layers.25.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
175
- "base_model.model.model.layers.25.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
176
- "base_model.model.model.layers.25.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
177
- "base_model.model.model.layers.25.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
178
- "base_model.model.model.layers.25.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
179
- "base_model.model.model.layers.26.input_layernorm.weight": "model-00003-of-00003.safetensors",
180
- "base_model.model.model.layers.26.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
181
- "base_model.model.model.layers.26.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
182
- "base_model.model.model.layers.26.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
183
- "base_model.model.model.layers.26.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
184
- "base_model.model.model.layers.26.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
185
- "base_model.model.model.layers.26.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
186
- "base_model.model.model.layers.26.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
187
- "base_model.model.model.layers.26.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
188
- "base_model.model.model.layers.27.input_layernorm.weight": "model-00003-of-00003.safetensors",
189
- "base_model.model.model.layers.27.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
190
- "base_model.model.model.layers.27.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
191
- "base_model.model.model.layers.27.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
192
- "base_model.model.model.layers.27.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
193
- "base_model.model.model.layers.27.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
194
- "base_model.model.model.layers.27.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
195
- "base_model.model.model.layers.27.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
196
- "base_model.model.model.layers.27.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
197
- "base_model.model.model.layers.28.input_layernorm.weight": "model-00003-of-00003.safetensors",
198
- "base_model.model.model.layers.28.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
199
- "base_model.model.model.layers.28.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
200
- "base_model.model.model.layers.28.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
201
- "base_model.model.model.layers.28.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
202
- "base_model.model.model.layers.28.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
203
- "base_model.model.model.layers.28.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
204
- "base_model.model.model.layers.28.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
205
- "base_model.model.model.layers.28.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
206
- "base_model.model.model.layers.29.input_layernorm.weight": "model-00003-of-00003.safetensors",
207
- "base_model.model.model.layers.29.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
208
- "base_model.model.model.layers.29.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
209
- "base_model.model.model.layers.29.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
210
- "base_model.model.model.layers.29.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
211
- "base_model.model.model.layers.29.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
212
- "base_model.model.model.layers.29.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
213
- "base_model.model.model.layers.29.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
214
- "base_model.model.model.layers.29.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
215
- "base_model.model.model.layers.3.input_layernorm.weight": "model-00001-of-00003.safetensors",
216
- "base_model.model.model.layers.3.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
217
- "base_model.model.model.layers.3.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
218
- "base_model.model.model.layers.3.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
219
- "base_model.model.model.layers.3.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
220
- "base_model.model.model.layers.3.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
221
- "base_model.model.model.layers.3.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
222
- "base_model.model.model.layers.3.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
223
- "base_model.model.model.layers.3.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
224
- "base_model.model.model.layers.30.input_layernorm.weight": "model-00003-of-00003.safetensors",
225
- "base_model.model.model.layers.30.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
226
- "base_model.model.model.layers.30.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
227
- "base_model.model.model.layers.30.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
228
- "base_model.model.model.layers.30.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
229
- "base_model.model.model.layers.30.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
230
- "base_model.model.model.layers.30.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
231
- "base_model.model.model.layers.30.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
232
- "base_model.model.model.layers.30.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
233
- "base_model.model.model.layers.31.input_layernorm.weight": "model-00003-of-00003.safetensors",
234
- "base_model.model.model.layers.31.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
235
- "base_model.model.model.layers.31.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
236
- "base_model.model.model.layers.31.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
237
- "base_model.model.model.layers.31.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
238
- "base_model.model.model.layers.31.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
239
- "base_model.model.model.layers.31.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
240
- "base_model.model.model.layers.31.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
241
- "base_model.model.model.layers.31.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
242
- "base_model.model.model.layers.4.input_layernorm.weight": "model-00001-of-00003.safetensors",
243
- "base_model.model.model.layers.4.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
244
- "base_model.model.model.layers.4.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
245
- "base_model.model.model.layers.4.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
246
- "base_model.model.model.layers.4.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
247
- "base_model.model.model.layers.4.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
248
- "base_model.model.model.layers.4.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
249
- "base_model.model.model.layers.4.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
250
- "base_model.model.model.layers.4.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
251
- "base_model.model.model.layers.5.input_layernorm.weight": "model-00001-of-00003.safetensors",
252
- "base_model.model.model.layers.5.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
253
- "base_model.model.model.layers.5.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
254
- "base_model.model.model.layers.5.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
255
- "base_model.model.model.layers.5.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
256
- "base_model.model.model.layers.5.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
257
- "base_model.model.model.layers.5.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
258
- "base_model.model.model.layers.5.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
259
- "base_model.model.model.layers.5.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
260
- "base_model.model.model.layers.6.input_layernorm.weight": "model-00001-of-00003.safetensors",
261
- "base_model.model.model.layers.6.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
262
- "base_model.model.model.layers.6.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
263
- "base_model.model.model.layers.6.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
264
- "base_model.model.model.layers.6.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
265
- "base_model.model.model.layers.6.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
266
- "base_model.model.model.layers.6.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
267
- "base_model.model.model.layers.6.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
268
- "base_model.model.model.layers.6.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
269
- "base_model.model.model.layers.7.input_layernorm.weight": "model-00001-of-00003.safetensors",
270
- "base_model.model.model.layers.7.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
271
- "base_model.model.model.layers.7.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
272
- "base_model.model.model.layers.7.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
273
- "base_model.model.model.layers.7.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
274
- "base_model.model.model.layers.7.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
275
- "base_model.model.model.layers.7.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
276
- "base_model.model.model.layers.7.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
277
- "base_model.model.model.layers.7.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
278
- "base_model.model.model.layers.8.input_layernorm.weight": "model-00001-of-00003.safetensors",
279
- "base_model.model.model.layers.8.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
280
- "base_model.model.model.layers.8.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
281
- "base_model.model.model.layers.8.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
282
- "base_model.model.model.layers.8.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
283
- "base_model.model.model.layers.8.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
284
- "base_model.model.model.layers.8.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
285
- "base_model.model.model.layers.8.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
286
- "base_model.model.model.layers.8.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
287
- "base_model.model.model.layers.9.input_layernorm.weight": "model-00001-of-00003.safetensors",
288
- "base_model.model.model.layers.9.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
289
- "base_model.model.model.layers.9.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
290
- "base_model.model.model.layers.9.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
291
- "base_model.model.model.layers.9.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
292
- "base_model.model.model.layers.9.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
293
- "base_model.model.model.layers.9.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
294
- "base_model.model.model.layers.9.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
295
- "base_model.model.model.layers.9.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
296
- "base_model.model.model.norm.weight": "model-00003-of-00003.safetensors"
297
  }
298
  }
 
3
  "total_size": 14483464192
4
  },
5
  "weight_map": {
6
+ "lm_head.weight": "model-00003-of-00003.safetensors",
7
+ "model.embed_tokens.weight": "model-00001-of-00003.safetensors",
8
+ "model.layers.0.input_layernorm.weight": "model-00001-of-00003.safetensors",
9
+ "model.layers.0.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
10
+ "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
11
+ "model.layers.0.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
12
+ "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
13
+ "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
14
+ "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
15
+ "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
16
+ "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
17
+ "model.layers.1.input_layernorm.weight": "model-00001-of-00003.safetensors",
18
+ "model.layers.1.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
19
+ "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
20
+ "model.layers.1.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
21
+ "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
22
+ "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
23
+ "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
24
+ "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
25
+ "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
26
+ "model.layers.10.input_layernorm.weight": "model-00002-of-00003.safetensors",
27
+ "model.layers.10.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
28
+ "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
29
+ "model.layers.10.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
30
+ "model.layers.10.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
31
+ "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
32
+ "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
33
+ "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
34
+ "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
35
+ "model.layers.11.input_layernorm.weight": "model-00002-of-00003.safetensors",
36
+ "model.layers.11.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
37
+ "model.layers.11.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
38
+ "model.layers.11.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
39
+ "model.layers.11.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
40
+ "model.layers.11.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
41
+ "model.layers.11.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
42
+ "model.layers.11.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
43
+ "model.layers.11.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
44
+ "model.layers.12.input_layernorm.weight": "model-00002-of-00003.safetensors",
45
+ "model.layers.12.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
46
+ "model.layers.12.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
47
+ "model.layers.12.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
48
+ "model.layers.12.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
49
+ "model.layers.12.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
50
+ "model.layers.12.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
51
+ "model.layers.12.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
52
+ "model.layers.12.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
53
+ "model.layers.13.input_layernorm.weight": "model-00002-of-00003.safetensors",
54
+ "model.layers.13.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
55
+ "model.layers.13.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
56
+ "model.layers.13.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
57
+ "model.layers.13.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
58
+ "model.layers.13.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
59
+ "model.layers.13.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
60
+ "model.layers.13.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
61
+ "model.layers.13.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
62
+ "model.layers.14.input_layernorm.weight": "model-00002-of-00003.safetensors",
63
+ "model.layers.14.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
64
+ "model.layers.14.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
65
+ "model.layers.14.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
66
+ "model.layers.14.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
67
+ "model.layers.14.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
68
+ "model.layers.14.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
69
+ "model.layers.14.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
70
+ "model.layers.14.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
71
+ "model.layers.15.input_layernorm.weight": "model-00002-of-00003.safetensors",
72
+ "model.layers.15.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
73
+ "model.layers.15.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
74
+ "model.layers.15.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
75
+ "model.layers.15.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
76
+ "model.layers.15.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
77
+ "model.layers.15.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
78
+ "model.layers.15.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
79
+ "model.layers.15.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
80
+ "model.layers.16.input_layernorm.weight": "model-00002-of-00003.safetensors",
81
+ "model.layers.16.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
82
+ "model.layers.16.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
83
+ "model.layers.16.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
84
+ "model.layers.16.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
85
+ "model.layers.16.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
86
+ "model.layers.16.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
87
+ "model.layers.16.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
88
+ "model.layers.16.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
89
+ "model.layers.17.input_layernorm.weight": "model-00002-of-00003.safetensors",
90
+ "model.layers.17.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
91
+ "model.layers.17.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
92
+ "model.layers.17.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
93
+ "model.layers.17.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
94
+ "model.layers.17.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
95
+ "model.layers.17.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
96
+ "model.layers.17.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
97
+ "model.layers.17.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
98
+ "model.layers.18.input_layernorm.weight": "model-00002-of-00003.safetensors",
99
+ "model.layers.18.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
100
+ "model.layers.18.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
101
+ "model.layers.18.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
102
+ "model.layers.18.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
103
+ "model.layers.18.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
104
+ "model.layers.18.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
105
+ "model.layers.18.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
106
+ "model.layers.18.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
107
+ "model.layers.19.input_layernorm.weight": "model-00002-of-00003.safetensors",
108
+ "model.layers.19.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
109
+ "model.layers.19.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
110
+ "model.layers.19.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
111
+ "model.layers.19.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
112
+ "model.layers.19.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
113
+ "model.layers.19.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
114
+ "model.layers.19.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
115
+ "model.layers.19.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
116
+ "model.layers.2.input_layernorm.weight": "model-00001-of-00003.safetensors",
117
+ "model.layers.2.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
118
+ "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
119
+ "model.layers.2.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
120
+ "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
121
+ "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
122
+ "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
123
+ "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
124
+ "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
125
+ "model.layers.20.input_layernorm.weight": "model-00002-of-00003.safetensors",
126
+ "model.layers.20.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
127
+ "model.layers.20.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
128
+ "model.layers.20.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
129
+ "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
130
+ "model.layers.20.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
131
+ "model.layers.20.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
132
+ "model.layers.20.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
133
+ "model.layers.20.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
134
+ "model.layers.21.input_layernorm.weight": "model-00002-of-00003.safetensors",
135
+ "model.layers.21.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
136
+ "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
137
+ "model.layers.21.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
138
+ "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
139
+ "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
140
+ "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
141
+ "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
142
+ "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
143
+ "model.layers.22.input_layernorm.weight": "model-00003-of-00003.safetensors",
144
+ "model.layers.22.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
145
+ "model.layers.22.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
146
+ "model.layers.22.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
147
+ "model.layers.22.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
148
+ "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
149
+ "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
150
+ "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
151
+ "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
152
+ "model.layers.23.input_layernorm.weight": "model-00003-of-00003.safetensors",
153
+ "model.layers.23.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
154
+ "model.layers.23.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
155
+ "model.layers.23.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
156
+ "model.layers.23.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
157
+ "model.layers.23.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
158
+ "model.layers.23.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
159
+ "model.layers.23.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
160
+ "model.layers.23.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
161
+ "model.layers.24.input_layernorm.weight": "model-00003-of-00003.safetensors",
162
+ "model.layers.24.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
163
+ "model.layers.24.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
164
+ "model.layers.24.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
165
+ "model.layers.24.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
166
+ "model.layers.24.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
167
+ "model.layers.24.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
168
+ "model.layers.24.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
169
+ "model.layers.24.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
170
+ "model.layers.25.input_layernorm.weight": "model-00003-of-00003.safetensors",
171
+ "model.layers.25.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
172
+ "model.layers.25.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
173
+ "model.layers.25.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
174
+ "model.layers.25.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
175
+ "model.layers.25.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
176
+ "model.layers.25.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
177
+ "model.layers.25.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
178
+ "model.layers.25.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
179
+ "model.layers.26.input_layernorm.weight": "model-00003-of-00003.safetensors",
180
+ "model.layers.26.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
181
+ "model.layers.26.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
182
+ "model.layers.26.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
183
+ "model.layers.26.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
184
+ "model.layers.26.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
185
+ "model.layers.26.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
186
+ "model.layers.26.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
187
+ "model.layers.26.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
188
+ "model.layers.27.input_layernorm.weight": "model-00003-of-00003.safetensors",
189
+ "model.layers.27.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
190
+ "model.layers.27.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
191
+ "model.layers.27.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
192
+ "model.layers.27.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
193
+ "model.layers.27.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
194
+ "model.layers.27.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
195
+ "model.layers.27.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
196
+ "model.layers.27.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
197
+ "model.layers.28.input_layernorm.weight": "model-00003-of-00003.safetensors",
198
+ "model.layers.28.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
199
+ "model.layers.28.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
200
+ "model.layers.28.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
201
+ "model.layers.28.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
202
+ "model.layers.28.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
203
+ "model.layers.28.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
204
+ "model.layers.28.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
205
+ "model.layers.28.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
206
+ "model.layers.29.input_layernorm.weight": "model-00003-of-00003.safetensors",
207
+ "model.layers.29.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
208
+ "model.layers.29.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
209
+ "model.layers.29.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
210
+ "model.layers.29.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
211
+ "model.layers.29.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
212
+ "model.layers.29.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
213
+ "model.layers.29.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
214
+ "model.layers.29.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
215
+ "model.layers.3.input_layernorm.weight": "model-00001-of-00003.safetensors",
216
+ "model.layers.3.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
217
+ "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
218
+ "model.layers.3.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
219
+ "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
220
+ "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
221
+ "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
222
+ "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
223
+ "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
224
+ "model.layers.30.input_layernorm.weight": "model-00003-of-00003.safetensors",
225
+ "model.layers.30.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
226
+ "model.layers.30.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
227
+ "model.layers.30.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
228
+ "model.layers.30.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
229
+ "model.layers.30.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
230
+ "model.layers.30.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
231
+ "model.layers.30.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
232
+ "model.layers.30.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
233
+ "model.layers.31.input_layernorm.weight": "model-00003-of-00003.safetensors",
234
+ "model.layers.31.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
235
+ "model.layers.31.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
236
+ "model.layers.31.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
237
+ "model.layers.31.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
238
+ "model.layers.31.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
239
+ "model.layers.31.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
240
+ "model.layers.31.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
241
+ "model.layers.31.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
242
+ "model.layers.4.input_layernorm.weight": "model-00001-of-00003.safetensors",
243
+ "model.layers.4.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
244
+ "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
245
+ "model.layers.4.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
246
+ "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
247
+ "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
248
+ "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
249
+ "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
250
+ "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
251
+ "model.layers.5.input_layernorm.weight": "model-00001-of-00003.safetensors",
252
+ "model.layers.5.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
253
+ "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
254
+ "model.layers.5.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
255
+ "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
256
+ "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
257
+ "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
258
+ "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
259
+ "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
260
+ "model.layers.6.input_layernorm.weight": "model-00001-of-00003.safetensors",
261
+ "model.layers.6.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
262
+ "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
263
+ "model.layers.6.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
264
+ "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
265
+ "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
266
+ "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
267
+ "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
268
+ "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
269
+ "model.layers.7.input_layernorm.weight": "model-00001-of-00003.safetensors",
270
+ "model.layers.7.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
271
+ "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
272
+ "model.layers.7.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
273
+ "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
274
+ "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
275
+ "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
276
+ "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
277
+ "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
278
+ "model.layers.8.input_layernorm.weight": "model-00001-of-00003.safetensors",
279
+ "model.layers.8.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
280
+ "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
281
+ "model.layers.8.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
282
+ "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
283
+ "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
284
+ "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
285
+ "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
286
+ "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
287
+ "model.layers.9.input_layernorm.weight": "model-00001-of-00003.safetensors",
288
+ "model.layers.9.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
289
+ "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
290
+ "model.layers.9.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
291
+ "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
292
+ "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
293
+ "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
294
+ "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
295
+ "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
296
+ "model.norm.weight": "model-00003-of-00003.safetensors"
297
  }
298
  }
sparsification_sftt.py CHANGED
@@ -551,6 +551,7 @@ class SparseMistralforCausalLM(MistralForCausalLM):
551
  m.mlp.dead_threshold = config.thresholds[idx]
552
  m.mlp.sparse_act_fn.set_new_threshold(m.mlp.dead_threshold)
553
  m.mlp.kill_sparse_swish_outputs = True
 
554
  if config.use_sparse_predictor:
555
  self.apply_sparse_predictor(init_svd=config.init_svd)
556
 
 
551
  m.mlp.dead_threshold = config.thresholds[idx]
552
  m.mlp.sparse_act_fn.set_new_threshold(m.mlp.dead_threshold)
553
  m.mlp.kill_sparse_swish_outputs = True
554
+ m.mlp.use_relu = config.use_relu
555
  if config.use_sparse_predictor:
556
  self.apply_sparse_predictor(init_svd=config.init_svd)
557
 
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0a26e098ad94464b8a24afc280fbfa799c23d2fd91b18354b170ceecf708f2e1
3
  size 6456
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c230c4a76feef1f1869d4277cd20574c6d2f4a752d464deb958409289a032f0
3
  size 6456