flozi00 commited on
Commit
bef6605
·
verified ·
1 Parent(s): 2df7630

Upload folder using huggingface_hub

Browse files
config.json CHANGED
@@ -19,7 +19,7 @@
19
  "rope_theta": 1000000.0,
20
  "sliding_window": 32768,
21
  "tie_word_embeddings": true,
22
- "torch_dtype": "bfloat16",
23
  "transformers_version": "4.41.2",
24
  "use_cache": false,
25
  "use_sliding_window": false,
 
19
  "rope_theta": 1000000.0,
20
  "sliding_window": 32768,
21
  "tie_word_embeddings": true,
22
+ "torch_dtype": "float32",
23
  "transformers_version": "4.41.2",
24
  "use_cache": false,
25
  "use_sliding_window": false,
generation_config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_from_model_config": true,
3
  "bos_token_id": 151643,
4
- "do_sample": false,
5
  "eos_token_id": [
6
  151645,
7
  151643
 
1
  {
2
  "_from_model_config": true,
3
  "bos_token_id": 151643,
4
+ "do_sample": true,
5
  "eos_token_id": [
6
  151645,
7
  151643
model-00001-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f66bfb42982b8c06c9bc25b8b99d14435f7a44a4827e6f1c4f4f62b47436fd6f
3
+ size 4996670464
model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c9c22855dcd0d98bb0e09d292039b2831909be91813061f03aa2692016df446
3
+ size 1178224960
model.safetensors.index.json ADDED
@@ -0,0 +1,345 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 6174857216
4
+ },
5
+ "weight_map": {
6
+ "model.embed_tokens.weight": "model-00001-of-00002.safetensors",
7
+ "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors",
8
+ "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
9
+ "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
10
+ "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
11
+ "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
12
+ "model.layers.0.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
13
+ "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
14
+ "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
15
+ "model.layers.0.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
16
+ "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
17
+ "model.layers.0.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
18
+ "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
19
+ "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors",
20
+ "model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
21
+ "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
22
+ "model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
23
+ "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
24
+ "model.layers.1.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
25
+ "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
26
+ "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
27
+ "model.layers.1.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
28
+ "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
29
+ "model.layers.1.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
30
+ "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
31
+ "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors",
32
+ "model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
33
+ "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
34
+ "model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
35
+ "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
36
+ "model.layers.10.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
37
+ "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
38
+ "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
39
+ "model.layers.10.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
40
+ "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
41
+ "model.layers.10.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
42
+ "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
43
+ "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors",
44
+ "model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
45
+ "model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
46
+ "model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
47
+ "model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
48
+ "model.layers.11.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
49
+ "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
50
+ "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
51
+ "model.layers.11.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
52
+ "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
53
+ "model.layers.11.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
54
+ "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
55
+ "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors",
56
+ "model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
57
+ "model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
58
+ "model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
59
+ "model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
60
+ "model.layers.12.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
61
+ "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
62
+ "model.layers.12.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
63
+ "model.layers.12.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
64
+ "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
65
+ "model.layers.12.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
66
+ "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
67
+ "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors",
68
+ "model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
69
+ "model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
70
+ "model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
71
+ "model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
72
+ "model.layers.13.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
73
+ "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
74
+ "model.layers.13.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
75
+ "model.layers.13.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
76
+ "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
77
+ "model.layers.13.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
78
+ "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
79
+ "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors",
80
+ "model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
81
+ "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
82
+ "model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
83
+ "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
84
+ "model.layers.14.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
85
+ "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
86
+ "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
87
+ "model.layers.14.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
88
+ "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
89
+ "model.layers.14.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
90
+ "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
91
+ "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors",
92
+ "model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
93
+ "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
94
+ "model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
95
+ "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
96
+ "model.layers.15.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
97
+ "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
98
+ "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
99
+ "model.layers.15.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
100
+ "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
101
+ "model.layers.15.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
102
+ "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
103
+ "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors",
104
+ "model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
105
+ "model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
106
+ "model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
107
+ "model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
108
+ "model.layers.16.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
109
+ "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
110
+ "model.layers.16.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
111
+ "model.layers.16.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
112
+ "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
113
+ "model.layers.16.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
114
+ "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
115
+ "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors",
116
+ "model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
117
+ "model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
118
+ "model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
119
+ "model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
120
+ "model.layers.17.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
121
+ "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
122
+ "model.layers.17.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
123
+ "model.layers.17.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
124
+ "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
125
+ "model.layers.17.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
126
+ "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
127
+ "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors",
128
+ "model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
129
+ "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
130
+ "model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
131
+ "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
132
+ "model.layers.18.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
133
+ "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
134
+ "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
135
+ "model.layers.18.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
136
+ "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
137
+ "model.layers.18.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
138
+ "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
139
+ "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors",
140
+ "model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
141
+ "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
142
+ "model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
143
+ "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
144
+ "model.layers.19.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
145
+ "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
146
+ "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
147
+ "model.layers.19.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
148
+ "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
149
+ "model.layers.19.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
150
+ "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
151
+ "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors",
152
+ "model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
153
+ "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
154
+ "model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
155
+ "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
156
+ "model.layers.2.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
157
+ "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
158
+ "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
159
+ "model.layers.2.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
160
+ "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
161
+ "model.layers.2.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
162
+ "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
163
+ "model.layers.20.input_layernorm.weight": "model-00001-of-00002.safetensors",
164
+ "model.layers.20.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
165
+ "model.layers.20.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
166
+ "model.layers.20.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
167
+ "model.layers.20.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
168
+ "model.layers.20.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
169
+ "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
170
+ "model.layers.20.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
171
+ "model.layers.20.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
172
+ "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
173
+ "model.layers.20.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
174
+ "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
175
+ "model.layers.21.input_layernorm.weight": "model-00002-of-00002.safetensors",
176
+ "model.layers.21.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
177
+ "model.layers.21.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
178
+ "model.layers.21.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
179
+ "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
180
+ "model.layers.21.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
181
+ "model.layers.21.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
182
+ "model.layers.21.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
183
+ "model.layers.21.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
184
+ "model.layers.21.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
185
+ "model.layers.21.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
186
+ "model.layers.21.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
187
+ "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors",
188
+ "model.layers.22.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
189
+ "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
190
+ "model.layers.22.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
191
+ "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
192
+ "model.layers.22.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
193
+ "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
194
+ "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
195
+ "model.layers.22.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
196
+ "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
197
+ "model.layers.22.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
198
+ "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
199
+ "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors",
200
+ "model.layers.23.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
201
+ "model.layers.23.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
202
+ "model.layers.23.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
203
+ "model.layers.23.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
204
+ "model.layers.23.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
205
+ "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
206
+ "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
207
+ "model.layers.23.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
208
+ "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
209
+ "model.layers.23.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
210
+ "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
211
+ "model.layers.24.input_layernorm.weight": "model-00002-of-00002.safetensors",
212
+ "model.layers.24.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
213
+ "model.layers.24.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
214
+ "model.layers.24.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
215
+ "model.layers.24.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
216
+ "model.layers.24.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
217
+ "model.layers.24.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
218
+ "model.layers.24.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
219
+ "model.layers.24.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
220
+ "model.layers.24.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
221
+ "model.layers.24.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
222
+ "model.layers.24.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
223
+ "model.layers.25.input_layernorm.weight": "model-00002-of-00002.safetensors",
224
+ "model.layers.25.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
225
+ "model.layers.25.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
226
+ "model.layers.25.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
227
+ "model.layers.25.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
228
+ "model.layers.25.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
229
+ "model.layers.25.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
230
+ "model.layers.25.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
231
+ "model.layers.25.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
232
+ "model.layers.25.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
233
+ "model.layers.25.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
234
+ "model.layers.25.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
235
+ "model.layers.26.input_layernorm.weight": "model-00002-of-00002.safetensors",
236
+ "model.layers.26.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
237
+ "model.layers.26.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
238
+ "model.layers.26.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
239
+ "model.layers.26.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
240
+ "model.layers.26.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
241
+ "model.layers.26.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
242
+ "model.layers.26.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
243
+ "model.layers.26.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
244
+ "model.layers.26.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
245
+ "model.layers.26.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
246
+ "model.layers.26.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
247
+ "model.layers.27.input_layernorm.weight": "model-00002-of-00002.safetensors",
248
+ "model.layers.27.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
249
+ "model.layers.27.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
250
+ "model.layers.27.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
251
+ "model.layers.27.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
252
+ "model.layers.27.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
253
+ "model.layers.27.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
254
+ "model.layers.27.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
255
+ "model.layers.27.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
256
+ "model.layers.27.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
257
+ "model.layers.27.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
258
+ "model.layers.27.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
259
+ "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors",
260
+ "model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
261
+ "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
262
+ "model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
263
+ "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
264
+ "model.layers.3.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
265
+ "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
266
+ "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
267
+ "model.layers.3.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
268
+ "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
269
+ "model.layers.3.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
270
+ "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
271
+ "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors",
272
+ "model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
273
+ "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
274
+ "model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
275
+ "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
276
+ "model.layers.4.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
277
+ "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
278
+ "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
279
+ "model.layers.4.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
280
+ "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
281
+ "model.layers.4.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
282
+ "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
283
+ "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors",
284
+ "model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
285
+ "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
286
+ "model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
287
+ "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
288
+ "model.layers.5.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
289
+ "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
290
+ "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
291
+ "model.layers.5.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
292
+ "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
293
+ "model.layers.5.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
294
+ "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
295
+ "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors",
296
+ "model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
297
+ "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
298
+ "model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
299
+ "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
300
+ "model.layers.6.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
301
+ "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
302
+ "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
303
+ "model.layers.6.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
304
+ "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
305
+ "model.layers.6.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
306
+ "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
307
+ "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors",
308
+ "model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
309
+ "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
310
+ "model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
311
+ "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
312
+ "model.layers.7.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
313
+ "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
314
+ "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
315
+ "model.layers.7.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
316
+ "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
317
+ "model.layers.7.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
318
+ "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
319
+ "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors",
320
+ "model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
321
+ "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
322
+ "model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
323
+ "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
324
+ "model.layers.8.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
325
+ "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
326
+ "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
327
+ "model.layers.8.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
328
+ "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
329
+ "model.layers.8.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
330
+ "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
331
+ "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors",
332
+ "model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
333
+ "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
334
+ "model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
335
+ "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
336
+ "model.layers.9.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
337
+ "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
338
+ "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
339
+ "model.layers.9.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
340
+ "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
341
+ "model.layers.9.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
342
+ "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
343
+ "model.norm.weight": "model-00002-of-00002.safetensors"
344
+ }
345
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a2af9d07fb61306ec347f800e5ef1d06cd08a760bc2a70df53fd800b38069f3
3
+ size 3094642562
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:386fcc8cc1089aade9450d86fb239ea3483f455fd2d78d8378645feecfec9d69
3
+ size 14244
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aca84d21747b494f2647f19153afa27bad629c2a874992b2bbe3f010aee01c0c
3
+ size 1064
trainer_state.json ADDED
@@ -0,0 +1,373 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 0.021066583570698138,
5
+ "eval_steps": 5000,
6
+ "global_step": 200,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.001053329178534907,
13
+ "grad_norm": 217.68748474121094,
14
+ "learning_rate": 5.263157894736842e-06,
15
+ "logits/chosen": -2.8389856815338135,
16
+ "logits/rejected": -2.8396875858306885,
17
+ "logps/chosen": -5.6633830070495605,
18
+ "logps/rejected": -6.1086931228637695,
19
+ "loss": 5.7387,
20
+ "odds_ratio_loss": 12.78534984588623,
21
+ "rewards/accuracies": 0.6208333373069763,
22
+ "rewards/chosen": -0.5663383603096008,
23
+ "rewards/margins": 0.04453102499246597,
24
+ "rewards/rejected": -0.6108693480491638,
25
+ "sft_loss": 4.460188388824463,
26
+ "step": 10
27
+ },
28
+ {
29
+ "epoch": 0.002106658357069814,
30
+ "grad_norm": 59.99885177612305,
31
+ "learning_rate": 1.0526315789473684e-05,
32
+ "logits/chosen": -3.0334506034851074,
33
+ "logits/rejected": -3.034069299697876,
34
+ "logps/chosen": -2.313845634460449,
35
+ "logps/rejected": -3.318277597427368,
36
+ "loss": 2.3687,
37
+ "odds_ratio_loss": 3.849175214767456,
38
+ "rewards/accuracies": 0.7562500238418579,
39
+ "rewards/chosen": -0.23138457536697388,
40
+ "rewards/margins": 0.10044321417808533,
41
+ "rewards/rejected": -0.3318277597427368,
42
+ "sft_loss": 1.9837348461151123,
43
+ "step": 20
44
+ },
45
+ {
46
+ "epoch": 0.0031599875356047207,
47
+ "grad_norm": 21.754234313964844,
48
+ "learning_rate": 1.5789473684210526e-05,
49
+ "logits/chosen": -3.226034641265869,
50
+ "logits/rejected": -3.2263522148132324,
51
+ "logps/chosen": -1.0712947845458984,
52
+ "logps/rejected": -2.3490335941314697,
53
+ "loss": 1.1116,
54
+ "odds_ratio_loss": 2.9476191997528076,
55
+ "rewards/accuracies": 0.8083333373069763,
56
+ "rewards/chosen": -0.10712946206331253,
57
+ "rewards/margins": 0.12777391076087952,
58
+ "rewards/rejected": -0.23490336537361145,
59
+ "sft_loss": 0.816817581653595,
60
+ "step": 30
61
+ },
62
+ {
63
+ "epoch": 0.004213316714139628,
64
+ "grad_norm": 7.782005310058594,
65
+ "learning_rate": 2.105263157894737e-05,
66
+ "logits/chosen": -3.4438676834106445,
67
+ "logits/rejected": -3.4442031383514404,
68
+ "logps/chosen": -0.8733291625976562,
69
+ "logps/rejected": -2.5004069805145264,
70
+ "loss": 0.914,
71
+ "odds_ratio_loss": 3.1110053062438965,
72
+ "rewards/accuracies": 0.8125,
73
+ "rewards/chosen": -0.08733292669057846,
74
+ "rewards/margins": 0.16270779073238373,
75
+ "rewards/rejected": -0.2500407099723816,
76
+ "sft_loss": 0.6028769016265869,
77
+ "step": 40
78
+ },
79
+ {
80
+ "epoch": 0.0052666458926745345,
81
+ "grad_norm": 9.219217300415039,
82
+ "learning_rate": 2.631578947368421e-05,
83
+ "logits/chosen": -3.4509921073913574,
84
+ "logits/rejected": -3.451155662536621,
85
+ "logps/chosen": -0.7789952158927917,
86
+ "logps/rejected": -2.3700900077819824,
87
+ "loss": 0.8189,
88
+ "odds_ratio_loss": 2.722813367843628,
89
+ "rewards/accuracies": 0.8062499761581421,
90
+ "rewards/chosen": -0.0778995230793953,
91
+ "rewards/margins": 0.1591094583272934,
92
+ "rewards/rejected": -0.23700900375843048,
93
+ "sft_loss": 0.5466489195823669,
94
+ "step": 50
95
+ },
96
+ {
97
+ "epoch": 0.006319975071209441,
98
+ "grad_norm": 7.359012603759766,
99
+ "learning_rate": 3.157894736842105e-05,
100
+ "logits/chosen": -3.5323359966278076,
101
+ "logits/rejected": -3.532480001449585,
102
+ "logps/chosen": -0.7335668802261353,
103
+ "logps/rejected": -2.5415773391723633,
104
+ "loss": 0.7747,
105
+ "odds_ratio_loss": 2.7141709327697754,
106
+ "rewards/accuracies": 0.800000011920929,
107
+ "rewards/chosen": -0.07335668057203293,
108
+ "rewards/margins": 0.18080104887485504,
109
+ "rewards/rejected": -0.2541577219963074,
110
+ "sft_loss": 0.5032956600189209,
111
+ "step": 60
112
+ },
113
+ {
114
+ "epoch": 0.007373304249744348,
115
+ "grad_norm": 16.543472290039062,
116
+ "learning_rate": 3.6842105263157895e-05,
117
+ "logits/chosen": -3.2315316200256348,
118
+ "logits/rejected": -3.2316715717315674,
119
+ "logps/chosen": -0.6829439997673035,
120
+ "logps/rejected": -2.482815980911255,
121
+ "loss": 0.7205,
122
+ "odds_ratio_loss": 2.6182825565338135,
123
+ "rewards/accuracies": 0.8374999761581421,
124
+ "rewards/chosen": -0.06829439848661423,
125
+ "rewards/margins": 0.17998717725276947,
126
+ "rewards/rejected": -0.2482815831899643,
127
+ "sft_loss": 0.4587023854255676,
128
+ "step": 70
129
+ },
130
+ {
131
+ "epoch": 0.008426633428279255,
132
+ "grad_norm": 5.6111907958984375,
133
+ "learning_rate": 4.210526315789474e-05,
134
+ "logits/chosen": -2.9154586791992188,
135
+ "logits/rejected": -2.9156744480133057,
136
+ "logps/chosen": -0.7533618807792664,
137
+ "logps/rejected": -2.669802665710449,
138
+ "loss": 0.7912,
139
+ "odds_ratio_loss": 2.688143730163574,
140
+ "rewards/accuracies": 0.8333333134651184,
141
+ "rewards/chosen": -0.07533618807792664,
142
+ "rewards/margins": 0.19164405763149261,
143
+ "rewards/rejected": -0.26698026061058044,
144
+ "sft_loss": 0.5223438739776611,
145
+ "step": 80
146
+ },
147
+ {
148
+ "epoch": 0.009479962606814161,
149
+ "grad_norm": 3.4394092559814453,
150
+ "learning_rate": 4.736842105263158e-05,
151
+ "logits/chosen": -3.0740275382995605,
152
+ "logits/rejected": -3.0744776725769043,
153
+ "logps/chosen": -0.6786951422691345,
154
+ "logps/rejected": -3.1611907482147217,
155
+ "loss": 0.7145,
156
+ "odds_ratio_loss": 2.766305446624756,
157
+ "rewards/accuracies": 0.8270833492279053,
158
+ "rewards/chosen": -0.06786951422691345,
159
+ "rewards/margins": 0.2482495754957199,
160
+ "rewards/rejected": -0.31611910462379456,
161
+ "sft_loss": 0.4378434419631958,
162
+ "step": 90
163
+ },
164
+ {
165
+ "epoch": 0.010533291785349069,
166
+ "grad_norm": 9.483699798583984,
167
+ "learning_rate": 5.263157894736842e-05,
168
+ "logits/chosen": -3.197289228439331,
169
+ "logits/rejected": -3.19758677482605,
170
+ "logps/chosen": -0.6806226372718811,
171
+ "logps/rejected": -3.0961530208587646,
172
+ "loss": 0.7172,
173
+ "odds_ratio_loss": 2.700289726257324,
174
+ "rewards/accuracies": 0.84375,
175
+ "rewards/chosen": -0.06806226074695587,
176
+ "rewards/margins": 0.24155308306217194,
177
+ "rewards/rejected": -0.3096153140068054,
178
+ "sft_loss": 0.44717150926589966,
179
+ "step": 100
180
+ },
181
+ {
182
+ "epoch": 0.011586620963883975,
183
+ "grad_norm": 7.847203731536865,
184
+ "learning_rate": 5.789473684210527e-05,
185
+ "logits/chosen": -3.01198673248291,
186
+ "logits/rejected": -3.0121328830718994,
187
+ "logps/chosen": -0.7313061356544495,
188
+ "logps/rejected": -3.1134066581726074,
189
+ "loss": 0.7701,
190
+ "odds_ratio_loss": 2.80288028717041,
191
+ "rewards/accuracies": 0.8416666388511658,
192
+ "rewards/chosen": -0.07313062250614166,
193
+ "rewards/margins": 0.2382100373506546,
194
+ "rewards/rejected": -0.31134068965911865,
195
+ "sft_loss": 0.489812970161438,
196
+ "step": 110
197
+ },
198
+ {
199
+ "epoch": 0.012639950142418883,
200
+ "grad_norm": 9.357610702514648,
201
+ "learning_rate": 6.31578947368421e-05,
202
+ "logits/chosen": -2.793389320373535,
203
+ "logits/rejected": -2.793593406677246,
204
+ "logps/chosen": -0.744411826133728,
205
+ "logps/rejected": -3.2574427127838135,
206
+ "loss": 0.781,
207
+ "odds_ratio_loss": 2.4383304119110107,
208
+ "rewards/accuracies": 0.8374999761581421,
209
+ "rewards/chosen": -0.07444119453430176,
210
+ "rewards/margins": 0.251303106546402,
211
+ "rewards/rejected": -0.32574430108070374,
212
+ "sft_loss": 0.5372061133384705,
213
+ "step": 120
214
+ },
215
+ {
216
+ "epoch": 0.013693279320953789,
217
+ "grad_norm": 4.926955699920654,
218
+ "learning_rate": 6.842105263157896e-05,
219
+ "logits/chosen": -2.829728603363037,
220
+ "logits/rejected": -2.829943895339966,
221
+ "logps/chosen": -0.5513553619384766,
222
+ "logps/rejected": -2.877117872238159,
223
+ "loss": 0.5817,
224
+ "odds_ratio_loss": 2.001354694366455,
225
+ "rewards/accuracies": 0.875,
226
+ "rewards/chosen": -0.055135536938905716,
227
+ "rewards/margins": 0.23257622122764587,
228
+ "rewards/rejected": -0.2877117693424225,
229
+ "sft_loss": 0.38156595826148987,
230
+ "step": 130
231
+ },
232
+ {
233
+ "epoch": 0.014746608499488697,
234
+ "grad_norm": 8.740342140197754,
235
+ "learning_rate": 7.368421052631579e-05,
236
+ "logits/chosen": -2.9767470359802246,
237
+ "logits/rejected": -2.9770071506500244,
238
+ "logps/chosen": -0.5641220211982727,
239
+ "logps/rejected": -4.819352149963379,
240
+ "loss": 0.589,
241
+ "odds_ratio_loss": 2.0714950561523438,
242
+ "rewards/accuracies": 0.8833333253860474,
243
+ "rewards/chosen": -0.05641220510005951,
244
+ "rewards/margins": 0.4255230724811554,
245
+ "rewards/rejected": -0.48193517327308655,
246
+ "sft_loss": 0.3818718194961548,
247
+ "step": 140
248
+ },
249
+ {
250
+ "epoch": 0.015799937678023603,
251
+ "grad_norm": 21.316011428833008,
252
+ "learning_rate": 7.894736842105263e-05,
253
+ "logits/chosen": -2.9424824714660645,
254
+ "logits/rejected": -2.942646026611328,
255
+ "logps/chosen": -0.6596536040306091,
256
+ "logps/rejected": -5.643968105316162,
257
+ "loss": 0.6873,
258
+ "odds_ratio_loss": 2.2379026412963867,
259
+ "rewards/accuracies": 0.8833333253860474,
260
+ "rewards/chosen": -0.06596536934375763,
261
+ "rewards/margins": 0.4984314739704132,
262
+ "rewards/rejected": -0.5643967986106873,
263
+ "sft_loss": 0.46346980333328247,
264
+ "step": 150
265
+ },
266
+ {
267
+ "epoch": 0.01685326685655851,
268
+ "grad_norm": 8.542074203491211,
269
+ "learning_rate": 8.421052631578948e-05,
270
+ "logits/chosen": -2.788573741912842,
271
+ "logits/rejected": -2.7887728214263916,
272
+ "logps/chosen": -0.5694094300270081,
273
+ "logps/rejected": -4.549073219299316,
274
+ "loss": 0.5953,
275
+ "odds_ratio_loss": 2.08962082862854,
276
+ "rewards/accuracies": 0.9083333611488342,
277
+ "rewards/chosen": -0.05694093927741051,
278
+ "rewards/margins": 0.3979664146900177,
279
+ "rewards/rejected": -0.4549073278903961,
280
+ "sft_loss": 0.3863413631916046,
281
+ "step": 160
282
+ },
283
+ {
284
+ "epoch": 0.017906596035093418,
285
+ "grad_norm": 10.086228370666504,
286
+ "learning_rate": 8.947368421052632e-05,
287
+ "logits/chosen": -2.747136354446411,
288
+ "logits/rejected": -2.7473561763763428,
289
+ "logps/chosen": -0.5511812567710876,
290
+ "logps/rejected": -4.04136848449707,
291
+ "loss": 0.5864,
292
+ "odds_ratio_loss": 2.00999116897583,
293
+ "rewards/accuracies": 0.8479166626930237,
294
+ "rewards/chosen": -0.0551181361079216,
295
+ "rewards/margins": 0.34901875257492065,
296
+ "rewards/rejected": -0.40413692593574524,
297
+ "sft_loss": 0.3853704035282135,
298
+ "step": 170
299
+ },
300
+ {
301
+ "epoch": 0.018959925213628322,
302
+ "grad_norm": 6.374107837677002,
303
+ "learning_rate": 9.473684210526316e-05,
304
+ "logits/chosen": -2.6502909660339355,
305
+ "logits/rejected": -2.650648593902588,
306
+ "logps/chosen": -0.5040014982223511,
307
+ "logps/rejected": -4.205493450164795,
308
+ "loss": 0.5236,
309
+ "odds_ratio_loss": 1.6211795806884766,
310
+ "rewards/accuracies": 0.9208333492279053,
311
+ "rewards/chosen": -0.050400152802467346,
312
+ "rewards/margins": 0.3701492249965668,
313
+ "rewards/rejected": -0.4205494225025177,
314
+ "sft_loss": 0.3614722490310669,
315
+ "step": 180
316
+ },
317
+ {
318
+ "epoch": 0.02001325439216323,
319
+ "grad_norm": 2.717724323272705,
320
+ "learning_rate": 0.0001,
321
+ "logits/chosen": -2.7677037715911865,
322
+ "logits/rejected": -2.7680699825286865,
323
+ "logps/chosen": -0.5834678411483765,
324
+ "logps/rejected": -4.945567607879639,
325
+ "loss": 0.6101,
326
+ "odds_ratio_loss": 2.1636292934417725,
327
+ "rewards/accuracies": 0.893750011920929,
328
+ "rewards/chosen": -0.05834679678082466,
329
+ "rewards/margins": 0.4362100064754486,
330
+ "rewards/rejected": -0.49455681443214417,
331
+ "sft_loss": 0.39376381039619446,
332
+ "step": 190
333
+ },
334
+ {
335
+ "epoch": 0.021066583570698138,
336
+ "grad_norm": 6.805058479309082,
337
+ "learning_rate": 0.00010526315789473683,
338
+ "logits/chosen": -2.722885847091675,
339
+ "logits/rejected": -2.723130226135254,
340
+ "logps/chosen": -0.6606806516647339,
341
+ "logps/rejected": -4.050003528594971,
342
+ "loss": 0.6941,
343
+ "odds_ratio_loss": 2.383312940597534,
344
+ "rewards/accuracies": 0.856249988079071,
345
+ "rewards/chosen": -0.06606806069612503,
346
+ "rewards/margins": 0.3389323055744171,
347
+ "rewards/rejected": -0.4050002992153168,
348
+ "sft_loss": 0.4557226896286011,
349
+ "step": 200
350
+ }
351
+ ],
352
+ "logging_steps": 10,
353
+ "max_steps": 9493,
354
+ "num_input_tokens_seen": 0,
355
+ "num_train_epochs": 1,
356
+ "save_steps": 100,
357
+ "stateful_callbacks": {
358
+ "TrainerControl": {
359
+ "args": {
360
+ "should_epoch_stop": false,
361
+ "should_evaluate": false,
362
+ "should_log": false,
363
+ "should_save": true,
364
+ "should_training_stop": false
365
+ },
366
+ "attributes": {}
367
+ }
368
+ },
369
+ "total_flos": 1.4746225175966515e+17,
370
+ "train_batch_size": 2,
371
+ "trial_name": null,
372
+ "trial_params": null
373
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e837a34ab4844742aedf3dd3e7300b434f2db8149439013ddd4cef5ab4cee401
3
+ size 5240