Upload folder using huggingface_hub
Browse files- config.json +1 -1
- generation_config.json +1 -1
- model-00001-of-00002.safetensors +3 -0
- model-00002-of-00002.safetensors +3 -0
- model.safetensors.index.json +345 -0
- optimizer.pt +3 -0
- rng_state.pth +3 -0
- scheduler.pt +3 -0
- trainer_state.json +373 -0
- training_args.bin +3 -0
config.json
CHANGED
@@ -19,7 +19,7 @@
|
|
19 |
"rope_theta": 1000000.0,
|
20 |
"sliding_window": 32768,
|
21 |
"tie_word_embeddings": true,
|
22 |
-
"torch_dtype": "
|
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":
|
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
|