Upload 15 files
Browse files- .gitattributes +1 -0
- added_tokens.json +24 -0
- chat_template.jinja +54 -0
- config.json +55 -0
- generation_config.json +6 -0
- merges.txt +0 -0
- model.safetensors +3 -0
- optimizer.pt +3 -0
- rng_state.pth +3 -0
- scheduler.pt +3 -0
- special_tokens_map.json +25 -0
- tokenizer.json +3 -0
- tokenizer_config.json +207 -0
- trainer_state.json +354 -0
- training_args.bin +3 -0
- vocab.json +0 -0
.gitattributes
CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
added_tokens.json
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"</tool_call>": 151658,
|
3 |
+
"<tool_call>": 151657,
|
4 |
+
"<|box_end|>": 151649,
|
5 |
+
"<|box_start|>": 151648,
|
6 |
+
"<|endoftext|>": 151643,
|
7 |
+
"<|file_sep|>": 151664,
|
8 |
+
"<|fim_middle|>": 151660,
|
9 |
+
"<|fim_pad|>": 151662,
|
10 |
+
"<|fim_prefix|>": 151659,
|
11 |
+
"<|fim_suffix|>": 151661,
|
12 |
+
"<|im_end|>": 151645,
|
13 |
+
"<|im_start|>": 151644,
|
14 |
+
"<|image_pad|>": 151655,
|
15 |
+
"<|object_ref_end|>": 151647,
|
16 |
+
"<|object_ref_start|>": 151646,
|
17 |
+
"<|quad_end|>": 151651,
|
18 |
+
"<|quad_start|>": 151650,
|
19 |
+
"<|repo_name|>": 151663,
|
20 |
+
"<|video_pad|>": 151656,
|
21 |
+
"<|vision_end|>": 151653,
|
22 |
+
"<|vision_pad|>": 151654,
|
23 |
+
"<|vision_start|>": 151652
|
24 |
+
}
|
chat_template.jinja
ADDED
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{%- if tools %}
|
2 |
+
{{- '<|im_start|>system\n' }}
|
3 |
+
{%- if messages[0]['role'] == 'system' %}
|
4 |
+
{{- messages[0]['content'] }}
|
5 |
+
{%- else %}
|
6 |
+
{{- 'You are a helpful assistant.' }}
|
7 |
+
{%- endif %}
|
8 |
+
{{- "\n\n# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
|
9 |
+
{%- for tool in tools %}
|
10 |
+
{{- "\n" }}
|
11 |
+
{{- tool | tojson }}
|
12 |
+
{%- endfor %}
|
13 |
+
{{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
|
14 |
+
{%- else %}
|
15 |
+
{%- if messages[0]['role'] == 'system' %}
|
16 |
+
{{- '<|im_start|>system\n' + messages[0]['content'] + '<|im_end|>\n' }}
|
17 |
+
{%- else %}
|
18 |
+
{{- '<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n' }}
|
19 |
+
{%- endif %}
|
20 |
+
{%- endif %}
|
21 |
+
{%- for message in messages %}
|
22 |
+
{%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %}
|
23 |
+
{{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
|
24 |
+
{%- elif message.role == "assistant" %}
|
25 |
+
{{- '<|im_start|>' + message.role }}
|
26 |
+
{%- if message.content %}
|
27 |
+
{{- '\n' + message.content }}
|
28 |
+
{%- endif %}
|
29 |
+
{%- for tool_call in message.tool_calls %}
|
30 |
+
{%- if tool_call.function is defined %}
|
31 |
+
{%- set tool_call = tool_call.function %}
|
32 |
+
{%- endif %}
|
33 |
+
{{- '\n<tool_call>\n{"name": "' }}
|
34 |
+
{{- tool_call.name }}
|
35 |
+
{{- '", "arguments": ' }}
|
36 |
+
{{- tool_call.arguments | tojson }}
|
37 |
+
{{- '}\n</tool_call>' }}
|
38 |
+
{%- endfor %}
|
39 |
+
{{- '<|im_end|>\n' }}
|
40 |
+
{%- elif message.role == "tool" %}
|
41 |
+
{%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %}
|
42 |
+
{{- '<|im_start|>user' }}
|
43 |
+
{%- endif %}
|
44 |
+
{{- '\n<tool_response>\n' }}
|
45 |
+
{{- message.content }}
|
46 |
+
{{- '\n</tool_response>' }}
|
47 |
+
{%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
|
48 |
+
{{- '<|im_end|>\n' }}
|
49 |
+
{%- endif %}
|
50 |
+
{%- endif %}
|
51 |
+
{%- endfor %}
|
52 |
+
{%- if add_generation_prompt %}
|
53 |
+
{{- '<|im_start|>assistant\n' }}
|
54 |
+
{%- endif %}
|
config.json
ADDED
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"architectures": [
|
3 |
+
"Qwen2ForCausalLM"
|
4 |
+
],
|
5 |
+
"attention_dropout": 0.0,
|
6 |
+
"bos_token_id": 151643,
|
7 |
+
"eos_token_id": 151643,
|
8 |
+
"hidden_act": "silu",
|
9 |
+
"hidden_size": 896,
|
10 |
+
"initializer_range": 0.02,
|
11 |
+
"intermediate_size": 4864,
|
12 |
+
"layer_types": [
|
13 |
+
"full_attention",
|
14 |
+
"full_attention",
|
15 |
+
"full_attention",
|
16 |
+
"full_attention",
|
17 |
+
"full_attention",
|
18 |
+
"full_attention",
|
19 |
+
"full_attention",
|
20 |
+
"full_attention",
|
21 |
+
"full_attention",
|
22 |
+
"full_attention",
|
23 |
+
"full_attention",
|
24 |
+
"full_attention",
|
25 |
+
"full_attention",
|
26 |
+
"full_attention",
|
27 |
+
"full_attention",
|
28 |
+
"full_attention",
|
29 |
+
"full_attention",
|
30 |
+
"full_attention",
|
31 |
+
"full_attention",
|
32 |
+
"full_attention",
|
33 |
+
"full_attention",
|
34 |
+
"full_attention",
|
35 |
+
"full_attention",
|
36 |
+
"full_attention"
|
37 |
+
],
|
38 |
+
"max_position_embeddings": 32768,
|
39 |
+
"max_window_layers": 24,
|
40 |
+
"model_type": "qwen2",
|
41 |
+
"num_attention_heads": 14,
|
42 |
+
"num_hidden_layers": 24,
|
43 |
+
"num_key_value_heads": 2,
|
44 |
+
"rms_norm_eps": 1e-06,
|
45 |
+
"rope_scaling": null,
|
46 |
+
"rope_theta": 1000000.0,
|
47 |
+
"sliding_window": null,
|
48 |
+
"tie_word_embeddings": true,
|
49 |
+
"torch_dtype": "float32",
|
50 |
+
"transformers_version": "4.53.1",
|
51 |
+
"use_cache": true,
|
52 |
+
"use_mrope": false,
|
53 |
+
"use_sliding_window": false,
|
54 |
+
"vocab_size": 151936
|
55 |
+
}
|
generation_config.json
ADDED
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"bos_token_id": 151643,
|
3 |
+
"eos_token_id": 151643,
|
4 |
+
"max_new_tokens": 2048,
|
5 |
+
"transformers_version": "4.53.1"
|
6 |
+
}
|
merges.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8c5c349cb1481ea81ee05358111d622fc330db9edffaabfaf4064cb4de225cdd
|
3 |
+
size 1976163472
|
optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:caff98e52a0fb7a67201751dff4db47c9f96f2f45f5c12f7f59d738a68612e9d
|
3 |
+
size 3952505739
|
rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bae999e3bf3b913b0513ba4e2510cbbca43afe2c42c2d5e578e390dfa199860f
|
3 |
+
size 14645
|
scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:37137fcce90cbae257becb508234162a0904803cee4f854ad6be5b47dfb6a127
|
3 |
+
size 1465
|
special_tokens_map.json
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"additional_special_tokens": [
|
3 |
+
"<|im_start|>",
|
4 |
+
"<|im_end|>",
|
5 |
+
"<|object_ref_start|>",
|
6 |
+
"<|object_ref_end|>",
|
7 |
+
"<|box_start|>",
|
8 |
+
"<|box_end|>",
|
9 |
+
"<|quad_start|>",
|
10 |
+
"<|quad_end|>",
|
11 |
+
"<|vision_start|>",
|
12 |
+
"<|vision_end|>",
|
13 |
+
"<|vision_pad|>",
|
14 |
+
"<|image_pad|>",
|
15 |
+
"<|video_pad|>"
|
16 |
+
],
|
17 |
+
"eos_token": "<|im_end|>",
|
18 |
+
"pad_token": {
|
19 |
+
"content": "<|endoftext|>",
|
20 |
+
"lstrip": false,
|
21 |
+
"normalized": false,
|
22 |
+
"rstrip": false,
|
23 |
+
"single_word": false
|
24 |
+
}
|
25 |
+
}
|
tokenizer.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9c5ae00e602b8860cbd784ba82a8aa14e8feecec692e7076590d014d7b7fdafa
|
3 |
+
size 11421896
|
tokenizer_config.json
ADDED
@@ -0,0 +1,207 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_bos_token": false,
|
3 |
+
"add_prefix_space": false,
|
4 |
+
"added_tokens_decoder": {
|
5 |
+
"151643": {
|
6 |
+
"content": "<|endoftext|>",
|
7 |
+
"lstrip": false,
|
8 |
+
"normalized": false,
|
9 |
+
"rstrip": false,
|
10 |
+
"single_word": false,
|
11 |
+
"special": true
|
12 |
+
},
|
13 |
+
"151644": {
|
14 |
+
"content": "<|im_start|>",
|
15 |
+
"lstrip": false,
|
16 |
+
"normalized": false,
|
17 |
+
"rstrip": false,
|
18 |
+
"single_word": false,
|
19 |
+
"special": true
|
20 |
+
},
|
21 |
+
"151645": {
|
22 |
+
"content": "<|im_end|>",
|
23 |
+
"lstrip": false,
|
24 |
+
"normalized": false,
|
25 |
+
"rstrip": false,
|
26 |
+
"single_word": false,
|
27 |
+
"special": true
|
28 |
+
},
|
29 |
+
"151646": {
|
30 |
+
"content": "<|object_ref_start|>",
|
31 |
+
"lstrip": false,
|
32 |
+
"normalized": false,
|
33 |
+
"rstrip": false,
|
34 |
+
"single_word": false,
|
35 |
+
"special": true
|
36 |
+
},
|
37 |
+
"151647": {
|
38 |
+
"content": "<|object_ref_end|>",
|
39 |
+
"lstrip": false,
|
40 |
+
"normalized": false,
|
41 |
+
"rstrip": false,
|
42 |
+
"single_word": false,
|
43 |
+
"special": true
|
44 |
+
},
|
45 |
+
"151648": {
|
46 |
+
"content": "<|box_start|>",
|
47 |
+
"lstrip": false,
|
48 |
+
"normalized": false,
|
49 |
+
"rstrip": false,
|
50 |
+
"single_word": false,
|
51 |
+
"special": true
|
52 |
+
},
|
53 |
+
"151649": {
|
54 |
+
"content": "<|box_end|>",
|
55 |
+
"lstrip": false,
|
56 |
+
"normalized": false,
|
57 |
+
"rstrip": false,
|
58 |
+
"single_word": false,
|
59 |
+
"special": true
|
60 |
+
},
|
61 |
+
"151650": {
|
62 |
+
"content": "<|quad_start|>",
|
63 |
+
"lstrip": false,
|
64 |
+
"normalized": false,
|
65 |
+
"rstrip": false,
|
66 |
+
"single_word": false,
|
67 |
+
"special": true
|
68 |
+
},
|
69 |
+
"151651": {
|
70 |
+
"content": "<|quad_end|>",
|
71 |
+
"lstrip": false,
|
72 |
+
"normalized": false,
|
73 |
+
"rstrip": false,
|
74 |
+
"single_word": false,
|
75 |
+
"special": true
|
76 |
+
},
|
77 |
+
"151652": {
|
78 |
+
"content": "<|vision_start|>",
|
79 |
+
"lstrip": false,
|
80 |
+
"normalized": false,
|
81 |
+
"rstrip": false,
|
82 |
+
"single_word": false,
|
83 |
+
"special": true
|
84 |
+
},
|
85 |
+
"151653": {
|
86 |
+
"content": "<|vision_end|>",
|
87 |
+
"lstrip": false,
|
88 |
+
"normalized": false,
|
89 |
+
"rstrip": false,
|
90 |
+
"single_word": false,
|
91 |
+
"special": true
|
92 |
+
},
|
93 |
+
"151654": {
|
94 |
+
"content": "<|vision_pad|>",
|
95 |
+
"lstrip": false,
|
96 |
+
"normalized": false,
|
97 |
+
"rstrip": false,
|
98 |
+
"single_word": false,
|
99 |
+
"special": true
|
100 |
+
},
|
101 |
+
"151655": {
|
102 |
+
"content": "<|image_pad|>",
|
103 |
+
"lstrip": false,
|
104 |
+
"normalized": false,
|
105 |
+
"rstrip": false,
|
106 |
+
"single_word": false,
|
107 |
+
"special": true
|
108 |
+
},
|
109 |
+
"151656": {
|
110 |
+
"content": "<|video_pad|>",
|
111 |
+
"lstrip": false,
|
112 |
+
"normalized": false,
|
113 |
+
"rstrip": false,
|
114 |
+
"single_word": false,
|
115 |
+
"special": true
|
116 |
+
},
|
117 |
+
"151657": {
|
118 |
+
"content": "<tool_call>",
|
119 |
+
"lstrip": false,
|
120 |
+
"normalized": false,
|
121 |
+
"rstrip": false,
|
122 |
+
"single_word": false,
|
123 |
+
"special": false
|
124 |
+
},
|
125 |
+
"151658": {
|
126 |
+
"content": "</tool_call>",
|
127 |
+
"lstrip": false,
|
128 |
+
"normalized": false,
|
129 |
+
"rstrip": false,
|
130 |
+
"single_word": false,
|
131 |
+
"special": false
|
132 |
+
},
|
133 |
+
"151659": {
|
134 |
+
"content": "<|fim_prefix|>",
|
135 |
+
"lstrip": false,
|
136 |
+
"normalized": false,
|
137 |
+
"rstrip": false,
|
138 |
+
"single_word": false,
|
139 |
+
"special": false
|
140 |
+
},
|
141 |
+
"151660": {
|
142 |
+
"content": "<|fim_middle|>",
|
143 |
+
"lstrip": false,
|
144 |
+
"normalized": false,
|
145 |
+
"rstrip": false,
|
146 |
+
"single_word": false,
|
147 |
+
"special": false
|
148 |
+
},
|
149 |
+
"151661": {
|
150 |
+
"content": "<|fim_suffix|>",
|
151 |
+
"lstrip": false,
|
152 |
+
"normalized": false,
|
153 |
+
"rstrip": false,
|
154 |
+
"single_word": false,
|
155 |
+
"special": false
|
156 |
+
},
|
157 |
+
"151662": {
|
158 |
+
"content": "<|fim_pad|>",
|
159 |
+
"lstrip": false,
|
160 |
+
"normalized": false,
|
161 |
+
"rstrip": false,
|
162 |
+
"single_word": false,
|
163 |
+
"special": false
|
164 |
+
},
|
165 |
+
"151663": {
|
166 |
+
"content": "<|repo_name|>",
|
167 |
+
"lstrip": false,
|
168 |
+
"normalized": false,
|
169 |
+
"rstrip": false,
|
170 |
+
"single_word": false,
|
171 |
+
"special": false
|
172 |
+
},
|
173 |
+
"151664": {
|
174 |
+
"content": "<|file_sep|>",
|
175 |
+
"lstrip": false,
|
176 |
+
"normalized": false,
|
177 |
+
"rstrip": false,
|
178 |
+
"single_word": false,
|
179 |
+
"special": false
|
180 |
+
}
|
181 |
+
},
|
182 |
+
"additional_special_tokens": [
|
183 |
+
"<|im_start|>",
|
184 |
+
"<|im_end|>",
|
185 |
+
"<|object_ref_start|>",
|
186 |
+
"<|object_ref_end|>",
|
187 |
+
"<|box_start|>",
|
188 |
+
"<|box_end|>",
|
189 |
+
"<|quad_start|>",
|
190 |
+
"<|quad_end|>",
|
191 |
+
"<|vision_start|>",
|
192 |
+
"<|vision_end|>",
|
193 |
+
"<|vision_pad|>",
|
194 |
+
"<|image_pad|>",
|
195 |
+
"<|video_pad|>"
|
196 |
+
],
|
197 |
+
"bos_token": null,
|
198 |
+
"clean_up_tokenization_spaces": false,
|
199 |
+
"eos_token": "<|im_end|>",
|
200 |
+
"errors": "replace",
|
201 |
+
"extra_special_tokens": {},
|
202 |
+
"model_max_length": 131072,
|
203 |
+
"pad_token": "<|endoftext|>",
|
204 |
+
"split_special_tokens": false,
|
205 |
+
"tokenizer_class": "Qwen2Tokenizer",
|
206 |
+
"unk_token": null
|
207 |
+
}
|
trainer_state.json
ADDED
@@ -0,0 +1,354 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_global_step": 3060,
|
3 |
+
"best_metric": 0.44437724351882935,
|
4 |
+
"best_model_checkpoint": "./sft_output_simpleQA\\checkpoint-3060",
|
5 |
+
"epoch": 5.0,
|
6 |
+
"eval_steps": 500,
|
7 |
+
"global_step": 15300,
|
8 |
+
"is_hyper_param_search": false,
|
9 |
+
"is_local_process_zero": true,
|
10 |
+
"is_world_process_zero": true,
|
11 |
+
"log_history": [
|
12 |
+
{
|
13 |
+
"epoch": 0.16339869281045752,
|
14 |
+
"grad_norm": 1.7589987516403198,
|
15 |
+
"learning_rate": 1.9347712418300655e-05,
|
16 |
+
"loss": 0.6565,
|
17 |
+
"mean_token_accuracy": 0.8448051940202713,
|
18 |
+
"num_tokens": 260778.0,
|
19 |
+
"step": 500
|
20 |
+
},
|
21 |
+
{
|
22 |
+
"epoch": 0.32679738562091504,
|
23 |
+
"grad_norm": 1.7202545404434204,
|
24 |
+
"learning_rate": 1.8694117647058824e-05,
|
25 |
+
"loss": 0.5658,
|
26 |
+
"mean_token_accuracy": 0.8607444787025451,
|
27 |
+
"num_tokens": 532143.0,
|
28 |
+
"step": 1000
|
29 |
+
},
|
30 |
+
{
|
31 |
+
"epoch": 0.49019607843137253,
|
32 |
+
"grad_norm": 0.5248209834098816,
|
33 |
+
"learning_rate": 1.8040522875816995e-05,
|
34 |
+
"loss": 0.4945,
|
35 |
+
"mean_token_accuracy": 0.8776222993135452,
|
36 |
+
"num_tokens": 794641.0,
|
37 |
+
"step": 1500
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.6535947712418301,
|
41 |
+
"grad_norm": 1.153851866722107,
|
42 |
+
"learning_rate": 1.7386928104575163e-05,
|
43 |
+
"loss": 0.4709,
|
44 |
+
"mean_token_accuracy": 0.8819570996761322,
|
45 |
+
"num_tokens": 1067466.0,
|
46 |
+
"step": 2000
|
47 |
+
},
|
48 |
+
{
|
49 |
+
"epoch": 0.8169934640522876,
|
50 |
+
"grad_norm": 2.172006607055664,
|
51 |
+
"learning_rate": 1.6733333333333335e-05,
|
52 |
+
"loss": 0.4283,
|
53 |
+
"mean_token_accuracy": 0.8932461794614792,
|
54 |
+
"num_tokens": 1332041.0,
|
55 |
+
"step": 2500
|
56 |
+
},
|
57 |
+
{
|
58 |
+
"epoch": 0.9803921568627451,
|
59 |
+
"grad_norm": 1.1672557592391968,
|
60 |
+
"learning_rate": 1.6079738562091506e-05,
|
61 |
+
"loss": 0.4424,
|
62 |
+
"mean_token_accuracy": 0.8902202717065811,
|
63 |
+
"num_tokens": 1591566.0,
|
64 |
+
"step": 3000
|
65 |
+
},
|
66 |
+
{
|
67 |
+
"epoch": 1.0,
|
68 |
+
"eval_loss": 0.44437724351882935,
|
69 |
+
"eval_mean_token_accuracy": 0.8896721144309219,
|
70 |
+
"eval_num_tokens": 1620476.0,
|
71 |
+
"eval_runtime": 17.1879,
|
72 |
+
"eval_samples_per_second": 22.225,
|
73 |
+
"eval_steps_per_second": 22.225,
|
74 |
+
"step": 3060
|
75 |
+
},
|
76 |
+
{
|
77 |
+
"epoch": 1.1437908496732025,
|
78 |
+
"grad_norm": 1.5445764064788818,
|
79 |
+
"learning_rate": 1.5426143790849674e-05,
|
80 |
+
"loss": 0.2976,
|
81 |
+
"mean_token_accuracy": 0.9238177012205124,
|
82 |
+
"num_tokens": 1847540.0,
|
83 |
+
"step": 3500
|
84 |
+
},
|
85 |
+
{
|
86 |
+
"epoch": 1.3071895424836601,
|
87 |
+
"grad_norm": 1.4802546501159668,
|
88 |
+
"learning_rate": 1.4772549019607844e-05,
|
89 |
+
"loss": 0.301,
|
90 |
+
"mean_token_accuracy": 0.9228217270374298,
|
91 |
+
"num_tokens": 2114882.0,
|
92 |
+
"step": 4000
|
93 |
+
},
|
94 |
+
{
|
95 |
+
"epoch": 1.4705882352941178,
|
96 |
+
"grad_norm": 1.7694408893585205,
|
97 |
+
"learning_rate": 1.4118954248366016e-05,
|
98 |
+
"loss": 0.275,
|
99 |
+
"mean_token_accuracy": 0.9289815596342087,
|
100 |
+
"num_tokens": 2383543.0,
|
101 |
+
"step": 4500
|
102 |
+
},
|
103 |
+
{
|
104 |
+
"epoch": 1.6339869281045751,
|
105 |
+
"grad_norm": 0.5041053295135498,
|
106 |
+
"learning_rate": 1.3465359477124185e-05,
|
107 |
+
"loss": 0.2656,
|
108 |
+
"mean_token_accuracy": 0.9312243936061859,
|
109 |
+
"num_tokens": 2655408.0,
|
110 |
+
"step": 5000
|
111 |
+
},
|
112 |
+
{
|
113 |
+
"epoch": 1.7973856209150327,
|
114 |
+
"grad_norm": 1.395111083984375,
|
115 |
+
"learning_rate": 1.2811764705882355e-05,
|
116 |
+
"loss": 0.277,
|
117 |
+
"mean_token_accuracy": 0.9283499264717102,
|
118 |
+
"num_tokens": 2909181.0,
|
119 |
+
"step": 5500
|
120 |
+
},
|
121 |
+
{
|
122 |
+
"epoch": 1.9607843137254903,
|
123 |
+
"grad_norm": 0.5925441980361938,
|
124 |
+
"learning_rate": 1.2158169934640525e-05,
|
125 |
+
"loss": 0.2716,
|
126 |
+
"mean_token_accuracy": 0.9297851510047913,
|
127 |
+
"num_tokens": 3175738.0,
|
128 |
+
"step": 6000
|
129 |
+
},
|
130 |
+
{
|
131 |
+
"epoch": 2.0,
|
132 |
+
"eval_loss": 0.444907546043396,
|
133 |
+
"eval_mean_token_accuracy": 0.8944824228736119,
|
134 |
+
"eval_num_tokens": 3240952.0,
|
135 |
+
"eval_runtime": 16.6457,
|
136 |
+
"eval_samples_per_second": 22.949,
|
137 |
+
"eval_steps_per_second": 22.949,
|
138 |
+
"step": 6120
|
139 |
+
},
|
140 |
+
{
|
141 |
+
"epoch": 2.1241830065359477,
|
142 |
+
"grad_norm": 0.823498010635376,
|
143 |
+
"learning_rate": 1.1504575163398695e-05,
|
144 |
+
"loss": 0.178,
|
145 |
+
"mean_token_accuracy": 0.9529191081523896,
|
146 |
+
"num_tokens": 3438042.0,
|
147 |
+
"step": 6500
|
148 |
+
},
|
149 |
+
{
|
150 |
+
"epoch": 2.287581699346405,
|
151 |
+
"grad_norm": 2.165581703186035,
|
152 |
+
"learning_rate": 1.0850980392156865e-05,
|
153 |
+
"loss": 0.1592,
|
154 |
+
"mean_token_accuracy": 0.9576994030475616,
|
155 |
+
"num_tokens": 3704561.0,
|
156 |
+
"step": 7000
|
157 |
+
},
|
158 |
+
{
|
159 |
+
"epoch": 2.450980392156863,
|
160 |
+
"grad_norm": 0.9315876960754395,
|
161 |
+
"learning_rate": 1.0197385620915034e-05,
|
162 |
+
"loss": 0.1613,
|
163 |
+
"mean_token_accuracy": 0.9569584859609603,
|
164 |
+
"num_tokens": 3962939.0,
|
165 |
+
"step": 7500
|
166 |
+
},
|
167 |
+
{
|
168 |
+
"epoch": 2.6143790849673203,
|
169 |
+
"grad_norm": 0.6542718410491943,
|
170 |
+
"learning_rate": 9.543790849673204e-06,
|
171 |
+
"loss": 0.1498,
|
172 |
+
"mean_token_accuracy": 0.9596330618858337,
|
173 |
+
"num_tokens": 4227835.0,
|
174 |
+
"step": 8000
|
175 |
+
},
|
176 |
+
{
|
177 |
+
"epoch": 2.7777777777777777,
|
178 |
+
"grad_norm": 2.1195223331451416,
|
179 |
+
"learning_rate": 8.890196078431374e-06,
|
180 |
+
"loss": 0.1571,
|
181 |
+
"mean_token_accuracy": 0.9582107998132706,
|
182 |
+
"num_tokens": 4490222.0,
|
183 |
+
"step": 8500
|
184 |
+
},
|
185 |
+
{
|
186 |
+
"epoch": 2.9411764705882355,
|
187 |
+
"grad_norm": 0.9135161638259888,
|
188 |
+
"learning_rate": 8.236601307189544e-06,
|
189 |
+
"loss": 0.1617,
|
190 |
+
"mean_token_accuracy": 0.9568131999969482,
|
191 |
+
"num_tokens": 4768317.0,
|
192 |
+
"step": 9000
|
193 |
+
},
|
194 |
+
{
|
195 |
+
"epoch": 3.0,
|
196 |
+
"eval_loss": 0.48607146739959717,
|
197 |
+
"eval_mean_token_accuracy": 0.8936509206032878,
|
198 |
+
"eval_num_tokens": 4861428.0,
|
199 |
+
"eval_runtime": 18.0359,
|
200 |
+
"eval_samples_per_second": 21.18,
|
201 |
+
"eval_steps_per_second": 21.18,
|
202 |
+
"step": 9180
|
203 |
+
},
|
204 |
+
{
|
205 |
+
"epoch": 3.104575163398693,
|
206 |
+
"grad_norm": 0.5220202803611755,
|
207 |
+
"learning_rate": 7.5830065359477136e-06,
|
208 |
+
"loss": 0.1193,
|
209 |
+
"mean_token_accuracy": 0.9686885585784912,
|
210 |
+
"num_tokens": 5033546.0,
|
211 |
+
"step": 9500
|
212 |
+
},
|
213 |
+
{
|
214 |
+
"epoch": 3.2679738562091503,
|
215 |
+
"grad_norm": 1.7922545671463013,
|
216 |
+
"learning_rate": 6.929411764705883e-06,
|
217 |
+
"loss": 0.0853,
|
218 |
+
"mean_token_accuracy": 0.9778412123918533,
|
219 |
+
"num_tokens": 5299950.0,
|
220 |
+
"step": 10000
|
221 |
+
},
|
222 |
+
{
|
223 |
+
"epoch": 3.431372549019608,
|
224 |
+
"grad_norm": 0.9012777209281921,
|
225 |
+
"learning_rate": 6.275816993464052e-06,
|
226 |
+
"loss": 0.0882,
|
227 |
+
"mean_token_accuracy": 0.9765507321357727,
|
228 |
+
"num_tokens": 5562491.0,
|
229 |
+
"step": 10500
|
230 |
+
},
|
231 |
+
{
|
232 |
+
"epoch": 3.5947712418300655,
|
233 |
+
"grad_norm": 0.7348634600639343,
|
234 |
+
"learning_rate": 5.622222222222222e-06,
|
235 |
+
"loss": 0.083,
|
236 |
+
"mean_token_accuracy": 0.9779657925367355,
|
237 |
+
"num_tokens": 5831882.0,
|
238 |
+
"step": 11000
|
239 |
+
},
|
240 |
+
{
|
241 |
+
"epoch": 3.758169934640523,
|
242 |
+
"grad_norm": 1.3759405612945557,
|
243 |
+
"learning_rate": 4.968627450980393e-06,
|
244 |
+
"loss": 0.0859,
|
245 |
+
"mean_token_accuracy": 0.9769166078567505,
|
246 |
+
"num_tokens": 6089238.0,
|
247 |
+
"step": 11500
|
248 |
+
},
|
249 |
+
{
|
250 |
+
"epoch": 3.9215686274509802,
|
251 |
+
"grad_norm": 1.0385228395462036,
|
252 |
+
"learning_rate": 4.315032679738563e-06,
|
253 |
+
"loss": 0.0905,
|
254 |
+
"mean_token_accuracy": 0.9758910204172134,
|
255 |
+
"num_tokens": 6358110.0,
|
256 |
+
"step": 12000
|
257 |
+
},
|
258 |
+
{
|
259 |
+
"epoch": 4.0,
|
260 |
+
"eval_loss": 0.5520662665367126,
|
261 |
+
"eval_mean_token_accuracy": 0.8913699647518977,
|
262 |
+
"eval_num_tokens": 6481904.0,
|
263 |
+
"eval_runtime": 18.076,
|
264 |
+
"eval_samples_per_second": 21.133,
|
265 |
+
"eval_steps_per_second": 21.133,
|
266 |
+
"step": 12240
|
267 |
+
},
|
268 |
+
{
|
269 |
+
"epoch": 4.084967320261438,
|
270 |
+
"grad_norm": 1.1012531518936157,
|
271 |
+
"learning_rate": 3.6614379084967324e-06,
|
272 |
+
"loss": 0.0645,
|
273 |
+
"mean_token_accuracy": 0.9829886084794999,
|
274 |
+
"num_tokens": 6621827.0,
|
275 |
+
"step": 12500
|
276 |
+
},
|
277 |
+
{
|
278 |
+
"epoch": 4.248366013071895,
|
279 |
+
"grad_norm": 1.51634681224823,
|
280 |
+
"learning_rate": 3.007843137254902e-06,
|
281 |
+
"loss": 0.0492,
|
282 |
+
"mean_token_accuracy": 0.987127070903778,
|
283 |
+
"num_tokens": 6878025.0,
|
284 |
+
"step": 13000
|
285 |
+
},
|
286 |
+
{
|
287 |
+
"epoch": 4.411764705882353,
|
288 |
+
"grad_norm": 0.771700382232666,
|
289 |
+
"learning_rate": 2.354248366013072e-06,
|
290 |
+
"loss": 0.0565,
|
291 |
+
"mean_token_accuracy": 0.9855971633195877,
|
292 |
+
"num_tokens": 7144101.0,
|
293 |
+
"step": 13500
|
294 |
+
},
|
295 |
+
{
|
296 |
+
"epoch": 4.57516339869281,
|
297 |
+
"grad_norm": 1.2798362970352173,
|
298 |
+
"learning_rate": 1.7006535947712418e-06,
|
299 |
+
"loss": 0.051,
|
300 |
+
"mean_token_accuracy": 0.9865847392082214,
|
301 |
+
"num_tokens": 7416906.0,
|
302 |
+
"step": 14000
|
303 |
+
},
|
304 |
+
{
|
305 |
+
"epoch": 4.738562091503268,
|
306 |
+
"grad_norm": 0.4460426867008209,
|
307 |
+
"learning_rate": 1.0470588235294118e-06,
|
308 |
+
"loss": 0.0511,
|
309 |
+
"mean_token_accuracy": 0.9866908140182495,
|
310 |
+
"num_tokens": 7684903.0,
|
311 |
+
"step": 14500
|
312 |
+
},
|
313 |
+
{
|
314 |
+
"epoch": 4.901960784313726,
|
315 |
+
"grad_norm": 0.559505045413971,
|
316 |
+
"learning_rate": 3.934640522875818e-07,
|
317 |
+
"loss": 0.051,
|
318 |
+
"mean_token_accuracy": 0.9866579355001449,
|
319 |
+
"num_tokens": 7945394.0,
|
320 |
+
"step": 15000
|
321 |
+
},
|
322 |
+
{
|
323 |
+
"epoch": 5.0,
|
324 |
+
"eval_loss": 0.6016601324081421,
|
325 |
+
"eval_mean_token_accuracy": 0.8893632412112821,
|
326 |
+
"eval_num_tokens": 8102380.0,
|
327 |
+
"eval_runtime": 18.1768,
|
328 |
+
"eval_samples_per_second": 21.016,
|
329 |
+
"eval_steps_per_second": 21.016,
|
330 |
+
"step": 15300
|
331 |
+
}
|
332 |
+
],
|
333 |
+
"logging_steps": 500,
|
334 |
+
"max_steps": 15300,
|
335 |
+
"num_input_tokens_seen": 0,
|
336 |
+
"num_train_epochs": 5,
|
337 |
+
"save_steps": 500,
|
338 |
+
"stateful_callbacks": {
|
339 |
+
"TrainerControl": {
|
340 |
+
"args": {
|
341 |
+
"should_epoch_stop": false,
|
342 |
+
"should_evaluate": false,
|
343 |
+
"should_log": false,
|
344 |
+
"should_save": true,
|
345 |
+
"should_training_stop": true
|
346 |
+
},
|
347 |
+
"attributes": {}
|
348 |
+
}
|
349 |
+
},
|
350 |
+
"total_flos": 1.739895902823936e+16,
|
351 |
+
"train_batch_size": 1,
|
352 |
+
"trial_name": null,
|
353 |
+
"trial_params": null
|
354 |
+
}
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5b7e3c225d81034256954fd32a3016c23609f6a0f0205ee13d321f749a45bdd0
|
3 |
+
size 6161
|
vocab.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|