Raiff1982 commited on
Commit
248d2f7
·
verified ·
1 Parent(s): 8e8a69f

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -40,3 +40,6 @@ newton/tokenizer.json filter=lfs diff=lfs merge=lfs -text
40
  davinci/checkpoint-500/tokenizer.json filter=lfs diff=lfs merge=lfs -text
41
  davinci/checkpoint-939/tokenizer.json filter=lfs diff=lfs merge=lfs -text
42
  davinci/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
 
 
40
  davinci/checkpoint-500/tokenizer.json filter=lfs diff=lfs merge=lfs -text
41
  davinci/checkpoint-939/tokenizer.json filter=lfs diff=lfs merge=lfs -text
42
  davinci/tokenizer.json filter=lfs diff=lfs merge=lfs -text
43
+ empathy/checkpoint-500/tokenizer.json filter=lfs diff=lfs merge=lfs -text
44
+ empathy/checkpoint-939/tokenizer.json filter=lfs diff=lfs merge=lfs -text
45
+ empathy/tokenizer.json filter=lfs diff=lfs merge=lfs -text
empathy/README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: meta-llama/Llama-3.1-8B-Instruct
3
+ library_name: peft
4
+ model_name: empathy
5
+ tags:
6
+ - base_model:adapter:meta-llama/Llama-3.1-8B-Instruct
7
+ - lora
8
+ - sft
9
+ - transformers
10
+ - trl
11
+ licence: license
12
+ pipeline_tag: text-generation
13
+ ---
14
+
15
+ # Model Card for empathy
16
+
17
+ This model is a fine-tuned version of [meta-llama/Llama-3.1-8B-Instruct](https://huggingface.co/meta-llama/Llama-3.1-8B-Instruct).
18
+ It has been trained using [TRL](https://github.com/huggingface/trl).
19
+
20
+ ## Quick start
21
+
22
+ ```python
23
+ from transformers import pipeline
24
+
25
+ question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
26
+ generator = pipeline("text-generation", model="None", device="cuda")
27
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
28
+ print(output["generated_text"])
29
+ ```
30
+
31
+ ## Training procedure
32
+
33
+
34
+
35
+
36
+
37
+ This model was trained with SFT.
38
+
39
+ ### Framework versions
40
+
41
+ - PEFT 0.18.1
42
+ - TRL: 0.29.0
43
+ - Transformers: 5.3.0
44
+ - Pytorch: 2.10.0
45
+ - Datasets: 4.6.1
46
+ - Tokenizers: 0.22.2
47
+
48
+ ## Citations
49
+
50
+
51
+
52
+ Cite TRL as:
53
+
54
+ ```bibtex
55
+ @software{vonwerra2020trl,
56
+ title = {{TRL: Transformers Reinforcement Learning}},
57
+ author = {von Werra, Leandro and Belkada, Younes and Tunstall, Lewis and Beeching, Edward and Thrush, Tristan and Lambert, Nathan and Huang, Shengyi and Rasul, Kashif and Gallouédec, Quentin},
58
+ license = {Apache-2.0},
59
+ url = {https://github.com/huggingface/trl},
60
+ year = {2020}
61
+ }
62
+ ```
empathy/adapter_config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "meta-llama/Llama-3.1-8B-Instruct",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 32,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.05,
22
+ "megatron_config": null,
23
+ "megatron_core": "megatron.core",
24
+ "modules_to_save": null,
25
+ "peft_type": "LORA",
26
+ "peft_version": "0.18.1",
27
+ "qalora_group_size": 16,
28
+ "r": 16,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "v_proj",
33
+ "o_proj",
34
+ "k_proj",
35
+ "q_proj"
36
+ ],
37
+ "target_parameters": null,
38
+ "task_type": "CAUSAL_LM",
39
+ "trainable_token_indices": null,
40
+ "use_dora": false,
41
+ "use_qalora": false,
42
+ "use_rslora": false
43
+ }
empathy/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae48e5ab42f5024f6e9ac24cbbc8e54d07a151740a4c6c5ea834fcaf78101d45
3
+ size 27297544
empathy/chat_template.jinja ADDED
@@ -0,0 +1,109 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {{- bos_token }}
2
+ {%- if custom_tools is defined %}
3
+ {%- set tools = custom_tools %}
4
+ {%- endif %}
5
+ {%- if not tools_in_user_message is defined %}
6
+ {%- set tools_in_user_message = true %}
7
+ {%- endif %}
8
+ {%- if not date_string is defined %}
9
+ {%- set date_string = "26 Jul 2024" %}
10
+ {%- endif %}
11
+ {%- if not tools is defined %}
12
+ {%- set tools = none %}
13
+ {%- endif %}
14
+
15
+ {#- This block extracts the system message, so we can slot it into the right place. #}
16
+ {%- if messages[0]['role'] == 'system' %}
17
+ {%- set system_message = messages[0]['content']|trim %}
18
+ {%- set messages = messages[1:] %}
19
+ {%- else %}
20
+ {%- set system_message = "" %}
21
+ {%- endif %}
22
+
23
+ {#- System message + builtin tools #}
24
+ {{- "<|start_header_id|>system<|end_header_id|>\n\n" }}
25
+ {%- if builtin_tools is defined or tools is not none %}
26
+ {{- "Environment: ipython\n" }}
27
+ {%- endif %}
28
+ {%- if builtin_tools is defined %}
29
+ {{- "Tools: " + builtin_tools | reject('equalto', 'code_interpreter') | join(", ") + "\n\n"}}
30
+ {%- endif %}
31
+ {{- "Cutting Knowledge Date: December 2023\n" }}
32
+ {{- "Today Date: " + date_string + "\n\n" }}
33
+ {%- if tools is not none and not tools_in_user_message %}
34
+ {{- "You have access to the following functions. To call a function, please respond with JSON for a function call." }}
35
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
36
+ {{- "Do not use variables.\n\n" }}
37
+ {%- for t in tools %}
38
+ {{- t | tojson(indent=4) }}
39
+ {{- "\n\n" }}
40
+ {%- endfor %}
41
+ {%- endif %}
42
+ {{- system_message }}
43
+ {{- "<|eot_id|>" }}
44
+
45
+ {#- Custom tools are passed in a user message with some extra guidance #}
46
+ {%- if tools_in_user_message and not tools is none %}
47
+ {#- Extract the first user message so we can plug it in here #}
48
+ {%- if messages | length != 0 %}
49
+ {%- set first_user_message = messages[0]['content']|trim %}
50
+ {%- set messages = messages[1:] %}
51
+ {%- else %}
52
+ {{- raise_exception("Cannot put tools in the first user message when there's no first user message!") }}
53
+ {%- endif %}
54
+ {{- '<|start_header_id|>user<|end_header_id|>\n\n' -}}
55
+ {{- "Given the following functions, please respond with a JSON for a function call " }}
56
+ {{- "with its proper arguments that best answers the given prompt.\n\n" }}
57
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
58
+ {{- "Do not use variables.\n\n" }}
59
+ {%- for t in tools %}
60
+ {{- t | tojson(indent=4) }}
61
+ {{- "\n\n" }}
62
+ {%- endfor %}
63
+ {{- first_user_message + "<|eot_id|>"}}
64
+ {%- endif %}
65
+
66
+ {%- for message in messages %}
67
+ {%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}
68
+ {{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'+ message['content'] | trim + '<|eot_id|>' }}
69
+ {%- elif 'tool_calls' in message %}
70
+ {%- if not message.tool_calls|length == 1 %}
71
+ {{- raise_exception("This model only supports single tool-calls at once!") }}
72
+ {%- endif %}
73
+ {%- set tool_call = message.tool_calls[0].function %}
74
+ {%- if builtin_tools is defined and tool_call.name in builtin_tools %}
75
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' -}}
76
+ {{- "<|python_tag|>" + tool_call.name + ".call(" }}
77
+ {%- for arg_name, arg_val in tool_call.arguments | items %}
78
+ {{- arg_name + '="' + arg_val + '"' }}
79
+ {%- if not loop.last %}
80
+ {{- ", " }}
81
+ {%- endif %}
82
+ {%- endfor %}
83
+ {{- ")" }}
84
+ {%- else %}
85
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' -}}
86
+ {{- '{"name": "' + tool_call.name + '", ' }}
87
+ {{- '"parameters": ' }}
88
+ {{- tool_call.arguments | tojson }}
89
+ {{- "}" }}
90
+ {%- endif %}
91
+ {%- if builtin_tools is defined %}
92
+ {#- This means we're in ipython mode #}
93
+ {{- "<|eom_id|>" }}
94
+ {%- else %}
95
+ {{- "<|eot_id|>" }}
96
+ {%- endif %}
97
+ {%- elif message.role == "tool" or message.role == "ipython" %}
98
+ {{- "<|start_header_id|>ipython<|end_header_id|>\n\n" }}
99
+ {%- if message.content is mapping or message.content is iterable %}
100
+ {{- message.content | tojson }}
101
+ {%- else %}
102
+ {{- message.content }}
103
+ {%- endif %}
104
+ {{- "<|eot_id|>" }}
105
+ {%- endif %}
106
+ {%- endfor %}
107
+ {%- if add_generation_prompt %}
108
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' }}
109
+ {%- endif %}
empathy/checkpoint-500/README.md ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: meta-llama/Llama-3.1-8B-Instruct
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:meta-llama/Llama-3.1-8B-Instruct
7
+ - lora
8
+ - sft
9
+ - transformers
10
+ - trl
11
+ ---
12
+
13
+ # Model Card for Model ID
14
+
15
+ <!-- Provide a quick summary of what the model is/does. -->
16
+
17
+
18
+
19
+ ## Model Details
20
+
21
+ ### Model Description
22
+
23
+ <!-- Provide a longer summary of what this model is. -->
24
+
25
+
26
+
27
+ - **Developed by:** [More Information Needed]
28
+ - **Funded by [optional]:** [More Information Needed]
29
+ - **Shared by [optional]:** [More Information Needed]
30
+ - **Model type:** [More Information Needed]
31
+ - **Language(s) (NLP):** [More Information Needed]
32
+ - **License:** [More Information Needed]
33
+ - **Finetuned from model [optional]:** [More Information Needed]
34
+
35
+ ### Model Sources [optional]
36
+
37
+ <!-- Provide the basic links for the model. -->
38
+
39
+ - **Repository:** [More Information Needed]
40
+ - **Paper [optional]:** [More Information Needed]
41
+ - **Demo [optional]:** [More Information Needed]
42
+
43
+ ## Uses
44
+
45
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
46
+
47
+ ### Direct Use
48
+
49
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
50
+
51
+ [More Information Needed]
52
+
53
+ ### Downstream Use [optional]
54
+
55
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
56
+
57
+ [More Information Needed]
58
+
59
+ ### Out-of-Scope Use
60
+
61
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
62
+
63
+ [More Information Needed]
64
+
65
+ ## Bias, Risks, and Limitations
66
+
67
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
68
+
69
+ [More Information Needed]
70
+
71
+ ### Recommendations
72
+
73
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
74
+
75
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
76
+
77
+ ## How to Get Started with the Model
78
+
79
+ Use the code below to get started with the model.
80
+
81
+ [More Information Needed]
82
+
83
+ ## Training Details
84
+
85
+ ### Training Data
86
+
87
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
88
+
89
+ [More Information Needed]
90
+
91
+ ### Training Procedure
92
+
93
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
94
+
95
+ #### Preprocessing [optional]
96
+
97
+ [More Information Needed]
98
+
99
+
100
+ #### Training Hyperparameters
101
+
102
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
103
+
104
+ #### Speeds, Sizes, Times [optional]
105
+
106
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
107
+
108
+ [More Information Needed]
109
+
110
+ ## Evaluation
111
+
112
+ <!-- This section describes the evaluation protocols and provides the results. -->
113
+
114
+ ### Testing Data, Factors & Metrics
115
+
116
+ #### Testing Data
117
+
118
+ <!-- This should link to a Dataset Card if possible. -->
119
+
120
+ [More Information Needed]
121
+
122
+ #### Factors
123
+
124
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
125
+
126
+ [More Information Needed]
127
+
128
+ #### Metrics
129
+
130
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
131
+
132
+ [More Information Needed]
133
+
134
+ ### Results
135
+
136
+ [More Information Needed]
137
+
138
+ #### Summary
139
+
140
+
141
+
142
+ ## Model Examination [optional]
143
+
144
+ <!-- Relevant interpretability work for the model goes here -->
145
+
146
+ [More Information Needed]
147
+
148
+ ## Environmental Impact
149
+
150
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
151
+
152
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
153
+
154
+ - **Hardware Type:** [More Information Needed]
155
+ - **Hours used:** [More Information Needed]
156
+ - **Cloud Provider:** [More Information Needed]
157
+ - **Compute Region:** [More Information Needed]
158
+ - **Carbon Emitted:** [More Information Needed]
159
+
160
+ ## Technical Specifications [optional]
161
+
162
+ ### Model Architecture and Objective
163
+
164
+ [More Information Needed]
165
+
166
+ ### Compute Infrastructure
167
+
168
+ [More Information Needed]
169
+
170
+ #### Hardware
171
+
172
+ [More Information Needed]
173
+
174
+ #### Software
175
+
176
+ [More Information Needed]
177
+
178
+ ## Citation [optional]
179
+
180
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
181
+
182
+ **BibTeX:**
183
+
184
+ [More Information Needed]
185
+
186
+ **APA:**
187
+
188
+ [More Information Needed]
189
+
190
+ ## Glossary [optional]
191
+
192
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
193
+
194
+ [More Information Needed]
195
+
196
+ ## More Information [optional]
197
+
198
+ [More Information Needed]
199
+
200
+ ## Model Card Authors [optional]
201
+
202
+ [More Information Needed]
203
+
204
+ ## Model Card Contact
205
+
206
+ [More Information Needed]
207
+ ### Framework versions
208
+
209
+ - PEFT 0.18.1
empathy/checkpoint-500/adapter_config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "meta-llama/Llama-3.1-8B-Instruct",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 32,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.05,
22
+ "megatron_config": null,
23
+ "megatron_core": "megatron.core",
24
+ "modules_to_save": null,
25
+ "peft_type": "LORA",
26
+ "peft_version": "0.18.1",
27
+ "qalora_group_size": 16,
28
+ "r": 16,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "v_proj",
33
+ "o_proj",
34
+ "k_proj",
35
+ "q_proj"
36
+ ],
37
+ "target_parameters": null,
38
+ "task_type": "CAUSAL_LM",
39
+ "trainable_token_indices": null,
40
+ "use_dora": false,
41
+ "use_qalora": false,
42
+ "use_rslora": false
43
+ }
empathy/checkpoint-500/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0b363fb47e7bc49ee535548078e136edbf60c8cbde25a51319be8e00c5b1deaf
3
+ size 27297544
empathy/checkpoint-500/chat_template.jinja ADDED
@@ -0,0 +1,109 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {{- bos_token }}
2
+ {%- if custom_tools is defined %}
3
+ {%- set tools = custom_tools %}
4
+ {%- endif %}
5
+ {%- if not tools_in_user_message is defined %}
6
+ {%- set tools_in_user_message = true %}
7
+ {%- endif %}
8
+ {%- if not date_string is defined %}
9
+ {%- set date_string = "26 Jul 2024" %}
10
+ {%- endif %}
11
+ {%- if not tools is defined %}
12
+ {%- set tools = none %}
13
+ {%- endif %}
14
+
15
+ {#- This block extracts the system message, so we can slot it into the right place. #}
16
+ {%- if messages[0]['role'] == 'system' %}
17
+ {%- set system_message = messages[0]['content']|trim %}
18
+ {%- set messages = messages[1:] %}
19
+ {%- else %}
20
+ {%- set system_message = "" %}
21
+ {%- endif %}
22
+
23
+ {#- System message + builtin tools #}
24
+ {{- "<|start_header_id|>system<|end_header_id|>\n\n" }}
25
+ {%- if builtin_tools is defined or tools is not none %}
26
+ {{- "Environment: ipython\n" }}
27
+ {%- endif %}
28
+ {%- if builtin_tools is defined %}
29
+ {{- "Tools: " + builtin_tools | reject('equalto', 'code_interpreter') | join(", ") + "\n\n"}}
30
+ {%- endif %}
31
+ {{- "Cutting Knowledge Date: December 2023\n" }}
32
+ {{- "Today Date: " + date_string + "\n\n" }}
33
+ {%- if tools is not none and not tools_in_user_message %}
34
+ {{- "You have access to the following functions. To call a function, please respond with JSON for a function call." }}
35
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
36
+ {{- "Do not use variables.\n\n" }}
37
+ {%- for t in tools %}
38
+ {{- t | tojson(indent=4) }}
39
+ {{- "\n\n" }}
40
+ {%- endfor %}
41
+ {%- endif %}
42
+ {{- system_message }}
43
+ {{- "<|eot_id|>" }}
44
+
45
+ {#- Custom tools are passed in a user message with some extra guidance #}
46
+ {%- if tools_in_user_message and not tools is none %}
47
+ {#- Extract the first user message so we can plug it in here #}
48
+ {%- if messages | length != 0 %}
49
+ {%- set first_user_message = messages[0]['content']|trim %}
50
+ {%- set messages = messages[1:] %}
51
+ {%- else %}
52
+ {{- raise_exception("Cannot put tools in the first user message when there's no first user message!") }}
53
+ {%- endif %}
54
+ {{- '<|start_header_id|>user<|end_header_id|>\n\n' -}}
55
+ {{- "Given the following functions, please respond with a JSON for a function call " }}
56
+ {{- "with its proper arguments that best answers the given prompt.\n\n" }}
57
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
58
+ {{- "Do not use variables.\n\n" }}
59
+ {%- for t in tools %}
60
+ {{- t | tojson(indent=4) }}
61
+ {{- "\n\n" }}
62
+ {%- endfor %}
63
+ {{- first_user_message + "<|eot_id|>"}}
64
+ {%- endif %}
65
+
66
+ {%- for message in messages %}
67
+ {%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}
68
+ {{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'+ message['content'] | trim + '<|eot_id|>' }}
69
+ {%- elif 'tool_calls' in message %}
70
+ {%- if not message.tool_calls|length == 1 %}
71
+ {{- raise_exception("This model only supports single tool-calls at once!") }}
72
+ {%- endif %}
73
+ {%- set tool_call = message.tool_calls[0].function %}
74
+ {%- if builtin_tools is defined and tool_call.name in builtin_tools %}
75
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' -}}
76
+ {{- "<|python_tag|>" + tool_call.name + ".call(" }}
77
+ {%- for arg_name, arg_val in tool_call.arguments | items %}
78
+ {{- arg_name + '="' + arg_val + '"' }}
79
+ {%- if not loop.last %}
80
+ {{- ", " }}
81
+ {%- endif %}
82
+ {%- endfor %}
83
+ {{- ")" }}
84
+ {%- else %}
85
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' -}}
86
+ {{- '{"name": "' + tool_call.name + '", ' }}
87
+ {{- '"parameters": ' }}
88
+ {{- tool_call.arguments | tojson }}
89
+ {{- "}" }}
90
+ {%- endif %}
91
+ {%- if builtin_tools is defined %}
92
+ {#- This means we're in ipython mode #}
93
+ {{- "<|eom_id|>" }}
94
+ {%- else %}
95
+ {{- "<|eot_id|>" }}
96
+ {%- endif %}
97
+ {%- elif message.role == "tool" or message.role == "ipython" %}
98
+ {{- "<|start_header_id|>ipython<|end_header_id|>\n\n" }}
99
+ {%- if message.content is mapping or message.content is iterable %}
100
+ {{- message.content | tojson }}
101
+ {%- else %}
102
+ {{- message.content }}
103
+ {%- endif %}
104
+ {{- "<|eot_id|>" }}
105
+ {%- endif %}
106
+ {%- endfor %}
107
+ {%- if add_generation_prompt %}
108
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' }}
109
+ {%- endif %}
empathy/checkpoint-500/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9cf1d1e0cabd9f52f8c0d4d73721bfc9bea7825fc4bde641425de27f1f77ccba
3
+ size 54745547
empathy/checkpoint-500/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6daf2865697b5429d29f0477ad615fdf3874eaecc852e74fe108011ff0eb9a9
3
+ size 14645
empathy/checkpoint-500/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:820927e41cd14e86356818cda9df54ef4ef3f4c3a4d4b7a28cbf4df629afd0fd
3
+ size 1465
empathy/checkpoint-500/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b9e4e7fb171f92fd137b777cc2714bf87d11576700a1dcd7a399e7bbe39537b
3
+ size 17209920
empathy/checkpoint-500/tokenizer_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "bos_token": "<|begin_of_text|>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "eos_token": "<|eot_id|>",
6
+ "is_local": false,
7
+ "model_input_names": [
8
+ "input_ids",
9
+ "attention_mask"
10
+ ],
11
+ "model_max_length": 131072,
12
+ "pad_token": "<|eot_id|>",
13
+ "tokenizer_class": "TokenizersBackend"
14
+ }
empathy/checkpoint-500/trainer_state.json ADDED
@@ -0,0 +1,534 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 1.5984,
6
+ "eval_steps": 500,
7
+ "global_step": 500,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "entropy": 2.7027451276779173,
14
+ "epoch": 0.032,
15
+ "grad_norm": 0.263671875,
16
+ "learning_rate": 6.206896551724138e-05,
17
+ "loss": 2.8736827850341795,
18
+ "mean_token_accuracy": 0.45954256504774094,
19
+ "num_tokens": 55882.0,
20
+ "step": 10
21
+ },
22
+ {
23
+ "entropy": 2.2726758420467377,
24
+ "epoch": 0.064,
25
+ "grad_norm": 0.2431640625,
26
+ "learning_rate": 0.00013103448275862068,
27
+ "loss": 2.4437068939208983,
28
+ "mean_token_accuracy": 0.5048861749470234,
29
+ "num_tokens": 111677.0,
30
+ "step": 20
31
+ },
32
+ {
33
+ "entropy": 1.816487193107605,
34
+ "epoch": 0.096,
35
+ "grad_norm": 0.275390625,
36
+ "learning_rate": 0.0002,
37
+ "loss": 1.7309745788574218,
38
+ "mean_token_accuracy": 0.6087440773844719,
39
+ "num_tokens": 167580.0,
40
+ "step": 30
41
+ },
42
+ {
43
+ "entropy": 1.1870245933532715,
44
+ "epoch": 0.128,
45
+ "grad_norm": 0.275390625,
46
+ "learning_rate": 0.0001978021978021978,
47
+ "loss": 1.1468750953674316,
48
+ "mean_token_accuracy": 0.7256016343832016,
49
+ "num_tokens": 223679.0,
50
+ "step": 40
51
+ },
52
+ {
53
+ "entropy": 0.8109098121523857,
54
+ "epoch": 0.16,
55
+ "grad_norm": 0.333984375,
56
+ "learning_rate": 0.00019560439560439562,
57
+ "loss": 0.7429761409759521,
58
+ "mean_token_accuracy": 0.8220725610852242,
59
+ "num_tokens": 279556.0,
60
+ "step": 50
61
+ },
62
+ {
63
+ "entropy": 0.5473626285791398,
64
+ "epoch": 0.192,
65
+ "grad_norm": 0.267578125,
66
+ "learning_rate": 0.00019340659340659342,
67
+ "loss": 0.4774918556213379,
68
+ "mean_token_accuracy": 0.8853456184267998,
69
+ "num_tokens": 335848.0,
70
+ "step": 60
71
+ },
72
+ {
73
+ "entropy": 0.35489378273487093,
74
+ "epoch": 0.224,
75
+ "grad_norm": 0.263671875,
76
+ "learning_rate": 0.00019120879120879122,
77
+ "loss": 0.2937335968017578,
78
+ "mean_token_accuracy": 0.9316855490207672,
79
+ "num_tokens": 392355.0,
80
+ "step": 70
81
+ },
82
+ {
83
+ "entropy": 0.25775772109627726,
84
+ "epoch": 0.256,
85
+ "grad_norm": 0.251953125,
86
+ "learning_rate": 0.00018901098901098903,
87
+ "loss": 0.21065135002136232,
88
+ "mean_token_accuracy": 0.9512736469507217,
89
+ "num_tokens": 448792.0,
90
+ "step": 80
91
+ },
92
+ {
93
+ "entropy": 0.20272715911269187,
94
+ "epoch": 0.288,
95
+ "grad_norm": 0.1953125,
96
+ "learning_rate": 0.00018681318681318683,
97
+ "loss": 0.1630323886871338,
98
+ "mean_token_accuracy": 0.9601437523961067,
99
+ "num_tokens": 505118.0,
100
+ "step": 90
101
+ },
102
+ {
103
+ "entropy": 0.15652943775057793,
104
+ "epoch": 0.32,
105
+ "grad_norm": 0.14453125,
106
+ "learning_rate": 0.00018461538461538463,
107
+ "loss": 0.13783388137817382,
108
+ "mean_token_accuracy": 0.9648320719599723,
109
+ "num_tokens": 562043.0,
110
+ "step": 100
111
+ },
112
+ {
113
+ "entropy": 0.13883790336549282,
114
+ "epoch": 0.352,
115
+ "grad_norm": 0.1494140625,
116
+ "learning_rate": 0.0001824175824175824,
117
+ "loss": 0.1205636978149414,
118
+ "mean_token_accuracy": 0.9679180562496186,
119
+ "num_tokens": 618572.0,
120
+ "step": 110
121
+ },
122
+ {
123
+ "entropy": 0.1294231442734599,
124
+ "epoch": 0.384,
125
+ "grad_norm": 0.142578125,
126
+ "learning_rate": 0.00018021978021978024,
127
+ "loss": 0.10652012825012207,
128
+ "mean_token_accuracy": 0.9696741059422493,
129
+ "num_tokens": 674953.0,
130
+ "step": 120
131
+ },
132
+ {
133
+ "entropy": 0.12372843902558088,
134
+ "epoch": 0.416,
135
+ "grad_norm": 0.091796875,
136
+ "learning_rate": 0.00017802197802197802,
137
+ "loss": 0.09909504652023315,
138
+ "mean_token_accuracy": 0.9700024798512459,
139
+ "num_tokens": 730476.0,
140
+ "step": 130
141
+ },
142
+ {
143
+ "entropy": 0.11392448395490647,
144
+ "epoch": 0.448,
145
+ "grad_norm": 0.1376953125,
146
+ "learning_rate": 0.00017582417582417582,
147
+ "loss": 0.09517329335212707,
148
+ "mean_token_accuracy": 0.9697632819414139,
149
+ "num_tokens": 786626.0,
150
+ "step": 140
151
+ },
152
+ {
153
+ "entropy": 0.10535773131996393,
154
+ "epoch": 0.48,
155
+ "grad_norm": 0.12451171875,
156
+ "learning_rate": 0.00017362637362637365,
157
+ "loss": 0.08859028220176697,
158
+ "mean_token_accuracy": 0.9710599616169929,
159
+ "num_tokens": 842780.0,
160
+ "step": 150
161
+ },
162
+ {
163
+ "entropy": 0.1028441235423088,
164
+ "epoch": 0.512,
165
+ "grad_norm": 0.1455078125,
166
+ "learning_rate": 0.00017142857142857143,
167
+ "loss": 0.08795538544654846,
168
+ "mean_token_accuracy": 0.9719248920679092,
169
+ "num_tokens": 898828.0,
170
+ "step": 160
171
+ },
172
+ {
173
+ "entropy": 0.09524063598364592,
174
+ "epoch": 0.544,
175
+ "grad_norm": 0.17578125,
176
+ "learning_rate": 0.00016923076923076923,
177
+ "loss": 0.0860197126865387,
178
+ "mean_token_accuracy": 0.9718463212251663,
179
+ "num_tokens": 955109.0,
180
+ "step": 170
181
+ },
182
+ {
183
+ "entropy": 0.09921295028179884,
184
+ "epoch": 0.576,
185
+ "grad_norm": 0.11328125,
186
+ "learning_rate": 0.00016703296703296706,
187
+ "loss": 0.08638875484466553,
188
+ "mean_token_accuracy": 0.9711567848920822,
189
+ "num_tokens": 1010735.0,
190
+ "step": 180
191
+ },
192
+ {
193
+ "entropy": 0.09508467447012663,
194
+ "epoch": 0.608,
195
+ "grad_norm": 0.10888671875,
196
+ "learning_rate": 0.00016483516483516484,
197
+ "loss": 0.08325063586235046,
198
+ "mean_token_accuracy": 0.9720648273825645,
199
+ "num_tokens": 1067222.0,
200
+ "step": 190
201
+ },
202
+ {
203
+ "entropy": 0.09094681274145841,
204
+ "epoch": 0.64,
205
+ "grad_norm": 0.0927734375,
206
+ "learning_rate": 0.00016263736263736264,
207
+ "loss": 0.0826878547668457,
208
+ "mean_token_accuracy": 0.9720210924744606,
209
+ "num_tokens": 1122748.0,
210
+ "step": 200
211
+ },
212
+ {
213
+ "entropy": 0.09381346199661493,
214
+ "epoch": 0.672,
215
+ "grad_norm": 0.0634765625,
216
+ "learning_rate": 0.00016043956043956044,
217
+ "loss": 0.08256186246871948,
218
+ "mean_token_accuracy": 0.9720484882593154,
219
+ "num_tokens": 1178643.0,
220
+ "step": 210
221
+ },
222
+ {
223
+ "entropy": 0.09017684925347566,
224
+ "epoch": 0.704,
225
+ "grad_norm": 0.09375,
226
+ "learning_rate": 0.00015824175824175824,
227
+ "loss": 0.08275171518325805,
228
+ "mean_token_accuracy": 0.9719254687428475,
229
+ "num_tokens": 1234871.0,
230
+ "step": 220
231
+ },
232
+ {
233
+ "entropy": 0.08761313948780299,
234
+ "epoch": 0.736,
235
+ "grad_norm": 0.119140625,
236
+ "learning_rate": 0.00015604395604395605,
237
+ "loss": 0.08114268779754638,
238
+ "mean_token_accuracy": 0.9720645502209664,
239
+ "num_tokens": 1290368.0,
240
+ "step": 230
241
+ },
242
+ {
243
+ "entropy": 0.08477144911885262,
244
+ "epoch": 0.768,
245
+ "grad_norm": 0.08251953125,
246
+ "learning_rate": 0.00015384615384615385,
247
+ "loss": 0.07812036871910095,
248
+ "mean_token_accuracy": 0.9725761353969574,
249
+ "num_tokens": 1347259.0,
250
+ "step": 240
251
+ },
252
+ {
253
+ "entropy": 0.0840575411915779,
254
+ "epoch": 0.8,
255
+ "grad_norm": 0.10595703125,
256
+ "learning_rate": 0.00015164835164835165,
257
+ "loss": 0.07779947519302369,
258
+ "mean_token_accuracy": 0.9721481159329415,
259
+ "num_tokens": 1404092.0,
260
+ "step": 250
261
+ },
262
+ {
263
+ "entropy": 0.08560035470873117,
264
+ "epoch": 0.832,
265
+ "grad_norm": 0.0654296875,
266
+ "learning_rate": 0.00014945054945054946,
267
+ "loss": 0.07757753729820252,
268
+ "mean_token_accuracy": 0.9719480901956559,
269
+ "num_tokens": 1460432.0,
270
+ "step": 260
271
+ },
272
+ {
273
+ "entropy": 0.08481895942240954,
274
+ "epoch": 0.864,
275
+ "grad_norm": 0.07080078125,
276
+ "learning_rate": 0.00014725274725274726,
277
+ "loss": 0.07894231081008911,
278
+ "mean_token_accuracy": 0.9720895797014236,
279
+ "num_tokens": 1516153.0,
280
+ "step": 270
281
+ },
282
+ {
283
+ "entropy": 0.08592585939913988,
284
+ "epoch": 0.896,
285
+ "grad_norm": 0.08251953125,
286
+ "learning_rate": 0.00014505494505494506,
287
+ "loss": 0.07765401601791382,
288
+ "mean_token_accuracy": 0.973065185546875,
289
+ "num_tokens": 1572583.0,
290
+ "step": 280
291
+ },
292
+ {
293
+ "entropy": 0.0848662929609418,
294
+ "epoch": 0.928,
295
+ "grad_norm": 0.1103515625,
296
+ "learning_rate": 0.00014285714285714287,
297
+ "loss": 0.07779905796051026,
298
+ "mean_token_accuracy": 0.9732325404882431,
299
+ "num_tokens": 1628569.0,
300
+ "step": 290
301
+ },
302
+ {
303
+ "entropy": 0.08204772900789976,
304
+ "epoch": 0.96,
305
+ "grad_norm": 0.059326171875,
306
+ "learning_rate": 0.00014065934065934067,
307
+ "loss": 0.07686090469360352,
308
+ "mean_token_accuracy": 0.9732487082481385,
309
+ "num_tokens": 1684838.0,
310
+ "step": 300
311
+ },
312
+ {
313
+ "entropy": 0.08476632833480835,
314
+ "epoch": 0.992,
315
+ "grad_norm": 0.10791015625,
316
+ "learning_rate": 0.00013846153846153847,
317
+ "loss": 0.076432603597641,
318
+ "mean_token_accuracy": 0.9719811454415321,
319
+ "num_tokens": 1741902.0,
320
+ "step": 310
321
+ },
322
+ {
323
+ "entropy": 0.07945429298438524,
324
+ "epoch": 1.0224,
325
+ "grad_norm": 0.0634765625,
326
+ "learning_rate": 0.00013626373626373628,
327
+ "loss": 0.07495068311691284,
328
+ "mean_token_accuracy": 0.9731942399551994,
329
+ "num_tokens": 1795784.0,
330
+ "step": 320
331
+ },
332
+ {
333
+ "entropy": 0.0806297117844224,
334
+ "epoch": 1.0544,
335
+ "grad_norm": 0.076171875,
336
+ "learning_rate": 0.00013406593406593405,
337
+ "loss": 0.07468733191490173,
338
+ "mean_token_accuracy": 0.9737721592187881,
339
+ "num_tokens": 1852266.0,
340
+ "step": 330
341
+ },
342
+ {
343
+ "entropy": 0.07759423423558473,
344
+ "epoch": 1.0864,
345
+ "grad_norm": 0.07763671875,
346
+ "learning_rate": 0.00013186813186813188,
347
+ "loss": 0.07366250157356262,
348
+ "mean_token_accuracy": 0.9733460173010826,
349
+ "num_tokens": 1908148.0,
350
+ "step": 340
351
+ },
352
+ {
353
+ "entropy": 0.08119724169373513,
354
+ "epoch": 1.1184,
355
+ "grad_norm": 0.050537109375,
356
+ "learning_rate": 0.0001296703296703297,
357
+ "loss": 0.07312369346618652,
358
+ "mean_token_accuracy": 0.97357087880373,
359
+ "num_tokens": 1964791.0,
360
+ "step": 350
361
+ },
362
+ {
363
+ "entropy": 0.07786643356084824,
364
+ "epoch": 1.1504,
365
+ "grad_norm": 0.04833984375,
366
+ "learning_rate": 0.00012747252747252746,
367
+ "loss": 0.07509608864784241,
368
+ "mean_token_accuracy": 0.9725616052746773,
369
+ "num_tokens": 2020416.0,
370
+ "step": 360
371
+ },
372
+ {
373
+ "entropy": 0.07941898722201586,
374
+ "epoch": 1.1824,
375
+ "grad_norm": 0.068359375,
376
+ "learning_rate": 0.00012527472527472527,
377
+ "loss": 0.0736158847808838,
378
+ "mean_token_accuracy": 0.9736046925187111,
379
+ "num_tokens": 2076384.0,
380
+ "step": 370
381
+ },
382
+ {
383
+ "entropy": 0.0761857332661748,
384
+ "epoch": 1.2144,
385
+ "grad_norm": 0.1337890625,
386
+ "learning_rate": 0.0001230769230769231,
387
+ "loss": 0.07236781120300292,
388
+ "mean_token_accuracy": 0.9739944905042648,
389
+ "num_tokens": 2132795.0,
390
+ "step": 380
391
+ },
392
+ {
393
+ "entropy": 0.07662795521318913,
394
+ "epoch": 1.2464,
395
+ "grad_norm": 0.1474609375,
396
+ "learning_rate": 0.00012087912087912087,
397
+ "loss": 0.07198388576507568,
398
+ "mean_token_accuracy": 0.9738625854253768,
399
+ "num_tokens": 2189096.0,
400
+ "step": 390
401
+ },
402
+ {
403
+ "entropy": 0.07697693575173617,
404
+ "epoch": 1.2784,
405
+ "grad_norm": 0.0634765625,
406
+ "learning_rate": 0.00011868131868131869,
407
+ "loss": 0.07262890338897705,
408
+ "mean_token_accuracy": 0.9728588461875916,
409
+ "num_tokens": 2245179.0,
410
+ "step": 400
411
+ },
412
+ {
413
+ "entropy": 0.07774264849722386,
414
+ "epoch": 1.3104,
415
+ "grad_norm": 0.062255859375,
416
+ "learning_rate": 0.0001164835164835165,
417
+ "loss": 0.07280476689338684,
418
+ "mean_token_accuracy": 0.9733426690101623,
419
+ "num_tokens": 2300662.0,
420
+ "step": 410
421
+ },
422
+ {
423
+ "entropy": 0.07713319007307291,
424
+ "epoch": 1.3424,
425
+ "grad_norm": 0.048583984375,
426
+ "learning_rate": 0.00011428571428571428,
427
+ "loss": 0.07291589379310608,
428
+ "mean_token_accuracy": 0.9731875821948052,
429
+ "num_tokens": 2356598.0,
430
+ "step": 420
431
+ },
432
+ {
433
+ "entropy": 0.07562742829322815,
434
+ "epoch": 1.3744,
435
+ "grad_norm": 0.060791015625,
436
+ "learning_rate": 0.0001120879120879121,
437
+ "loss": 0.0718300998210907,
438
+ "mean_token_accuracy": 0.9734191030263901,
439
+ "num_tokens": 2412737.0,
440
+ "step": 430
441
+ },
442
+ {
443
+ "entropy": 0.07663317285478115,
444
+ "epoch": 1.4064,
445
+ "grad_norm": 0.058349609375,
446
+ "learning_rate": 0.0001098901098901099,
447
+ "loss": 0.0707779049873352,
448
+ "mean_token_accuracy": 0.9742564350366593,
449
+ "num_tokens": 2469165.0,
450
+ "step": 440
451
+ },
452
+ {
453
+ "entropy": 0.07312421463429927,
454
+ "epoch": 1.4384000000000001,
455
+ "grad_norm": 0.0654296875,
456
+ "learning_rate": 0.0001076923076923077,
457
+ "loss": 0.07058998346328735,
458
+ "mean_token_accuracy": 0.9742231473326683,
459
+ "num_tokens": 2525741.0,
460
+ "step": 450
461
+ },
462
+ {
463
+ "entropy": 0.07522066552191972,
464
+ "epoch": 1.4704,
465
+ "grad_norm": 0.052978515625,
466
+ "learning_rate": 0.0001054945054945055,
467
+ "loss": 0.07139057517051697,
468
+ "mean_token_accuracy": 0.9742771983146667,
469
+ "num_tokens": 2582350.0,
470
+ "step": 460
471
+ },
472
+ {
473
+ "entropy": 0.07417810596525669,
474
+ "epoch": 1.5024,
475
+ "grad_norm": 0.04443359375,
476
+ "learning_rate": 0.00010329670329670331,
477
+ "loss": 0.0715557336807251,
478
+ "mean_token_accuracy": 0.9731055706739425,
479
+ "num_tokens": 2638763.0,
480
+ "step": 470
481
+ },
482
+ {
483
+ "entropy": 0.07531943805515766,
484
+ "epoch": 1.5344,
485
+ "grad_norm": 0.08349609375,
486
+ "learning_rate": 0.0001010989010989011,
487
+ "loss": 0.07040765285491943,
488
+ "mean_token_accuracy": 0.9743428856134415,
489
+ "num_tokens": 2695023.0,
490
+ "step": 480
491
+ },
492
+ {
493
+ "entropy": 0.0734874278306961,
494
+ "epoch": 1.5664,
495
+ "grad_norm": 0.06591796875,
496
+ "learning_rate": 9.89010989010989e-05,
497
+ "loss": 0.07134199142456055,
498
+ "mean_token_accuracy": 0.9733200684189797,
499
+ "num_tokens": 2751169.0,
500
+ "step": 490
501
+ },
502
+ {
503
+ "entropy": 0.07535444647073745,
504
+ "epoch": 1.5984,
505
+ "grad_norm": 0.05615234375,
506
+ "learning_rate": 9.670329670329671e-05,
507
+ "loss": 0.07132225036621094,
508
+ "mean_token_accuracy": 0.9734165251255036,
509
+ "num_tokens": 2807473.0,
510
+ "step": 500
511
+ }
512
+ ],
513
+ "logging_steps": 10,
514
+ "max_steps": 939,
515
+ "num_input_tokens_seen": 0,
516
+ "num_train_epochs": 3,
517
+ "save_steps": 500,
518
+ "stateful_callbacks": {
519
+ "TrainerControl": {
520
+ "args": {
521
+ "should_epoch_stop": false,
522
+ "should_evaluate": false,
523
+ "should_log": false,
524
+ "should_save": true,
525
+ "should_training_stop": false
526
+ },
527
+ "attributes": {}
528
+ }
529
+ },
530
+ "total_flos": 1.308143060299776e+17,
531
+ "train_batch_size": 2,
532
+ "trial_name": null,
533
+ "trial_params": null
534
+ }
empathy/checkpoint-500/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e64d4e636d5c0eb6e65352529689dee60289c387b4536f3e0b17f48b352ee585
3
+ size 5585
empathy/checkpoint-939/README.md ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: meta-llama/Llama-3.1-8B-Instruct
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:meta-llama/Llama-3.1-8B-Instruct
7
+ - lora
8
+ - sft
9
+ - transformers
10
+ - trl
11
+ ---
12
+
13
+ # Model Card for Model ID
14
+
15
+ <!-- Provide a quick summary of what the model is/does. -->
16
+
17
+
18
+
19
+ ## Model Details
20
+
21
+ ### Model Description
22
+
23
+ <!-- Provide a longer summary of what this model is. -->
24
+
25
+
26
+
27
+ - **Developed by:** [More Information Needed]
28
+ - **Funded by [optional]:** [More Information Needed]
29
+ - **Shared by [optional]:** [More Information Needed]
30
+ - **Model type:** [More Information Needed]
31
+ - **Language(s) (NLP):** [More Information Needed]
32
+ - **License:** [More Information Needed]
33
+ - **Finetuned from model [optional]:** [More Information Needed]
34
+
35
+ ### Model Sources [optional]
36
+
37
+ <!-- Provide the basic links for the model. -->
38
+
39
+ - **Repository:** [More Information Needed]
40
+ - **Paper [optional]:** [More Information Needed]
41
+ - **Demo [optional]:** [More Information Needed]
42
+
43
+ ## Uses
44
+
45
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
46
+
47
+ ### Direct Use
48
+
49
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
50
+
51
+ [More Information Needed]
52
+
53
+ ### Downstream Use [optional]
54
+
55
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
56
+
57
+ [More Information Needed]
58
+
59
+ ### Out-of-Scope Use
60
+
61
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
62
+
63
+ [More Information Needed]
64
+
65
+ ## Bias, Risks, and Limitations
66
+
67
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
68
+
69
+ [More Information Needed]
70
+
71
+ ### Recommendations
72
+
73
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
74
+
75
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
76
+
77
+ ## How to Get Started with the Model
78
+
79
+ Use the code below to get started with the model.
80
+
81
+ [More Information Needed]
82
+
83
+ ## Training Details
84
+
85
+ ### Training Data
86
+
87
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
88
+
89
+ [More Information Needed]
90
+
91
+ ### Training Procedure
92
+
93
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
94
+
95
+ #### Preprocessing [optional]
96
+
97
+ [More Information Needed]
98
+
99
+
100
+ #### Training Hyperparameters
101
+
102
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
103
+
104
+ #### Speeds, Sizes, Times [optional]
105
+
106
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
107
+
108
+ [More Information Needed]
109
+
110
+ ## Evaluation
111
+
112
+ <!-- This section describes the evaluation protocols and provides the results. -->
113
+
114
+ ### Testing Data, Factors & Metrics
115
+
116
+ #### Testing Data
117
+
118
+ <!-- This should link to a Dataset Card if possible. -->
119
+
120
+ [More Information Needed]
121
+
122
+ #### Factors
123
+
124
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
125
+
126
+ [More Information Needed]
127
+
128
+ #### Metrics
129
+
130
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
131
+
132
+ [More Information Needed]
133
+
134
+ ### Results
135
+
136
+ [More Information Needed]
137
+
138
+ #### Summary
139
+
140
+
141
+
142
+ ## Model Examination [optional]
143
+
144
+ <!-- Relevant interpretability work for the model goes here -->
145
+
146
+ [More Information Needed]
147
+
148
+ ## Environmental Impact
149
+
150
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
151
+
152
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
153
+
154
+ - **Hardware Type:** [More Information Needed]
155
+ - **Hours used:** [More Information Needed]
156
+ - **Cloud Provider:** [More Information Needed]
157
+ - **Compute Region:** [More Information Needed]
158
+ - **Carbon Emitted:** [More Information Needed]
159
+
160
+ ## Technical Specifications [optional]
161
+
162
+ ### Model Architecture and Objective
163
+
164
+ [More Information Needed]
165
+
166
+ ### Compute Infrastructure
167
+
168
+ [More Information Needed]
169
+
170
+ #### Hardware
171
+
172
+ [More Information Needed]
173
+
174
+ #### Software
175
+
176
+ [More Information Needed]
177
+
178
+ ## Citation [optional]
179
+
180
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
181
+
182
+ **BibTeX:**
183
+
184
+ [More Information Needed]
185
+
186
+ **APA:**
187
+
188
+ [More Information Needed]
189
+
190
+ ## Glossary [optional]
191
+
192
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
193
+
194
+ [More Information Needed]
195
+
196
+ ## More Information [optional]
197
+
198
+ [More Information Needed]
199
+
200
+ ## Model Card Authors [optional]
201
+
202
+ [More Information Needed]
203
+
204
+ ## Model Card Contact
205
+
206
+ [More Information Needed]
207
+ ### Framework versions
208
+
209
+ - PEFT 0.18.1
empathy/checkpoint-939/adapter_config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "meta-llama/Llama-3.1-8B-Instruct",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 32,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.05,
22
+ "megatron_config": null,
23
+ "megatron_core": "megatron.core",
24
+ "modules_to_save": null,
25
+ "peft_type": "LORA",
26
+ "peft_version": "0.18.1",
27
+ "qalora_group_size": 16,
28
+ "r": 16,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "v_proj",
33
+ "o_proj",
34
+ "k_proj",
35
+ "q_proj"
36
+ ],
37
+ "target_parameters": null,
38
+ "task_type": "CAUSAL_LM",
39
+ "trainable_token_indices": null,
40
+ "use_dora": false,
41
+ "use_qalora": false,
42
+ "use_rslora": false
43
+ }
empathy/checkpoint-939/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae48e5ab42f5024f6e9ac24cbbc8e54d07a151740a4c6c5ea834fcaf78101d45
3
+ size 27297544
empathy/checkpoint-939/chat_template.jinja ADDED
@@ -0,0 +1,109 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {{- bos_token }}
2
+ {%- if custom_tools is defined %}
3
+ {%- set tools = custom_tools %}
4
+ {%- endif %}
5
+ {%- if not tools_in_user_message is defined %}
6
+ {%- set tools_in_user_message = true %}
7
+ {%- endif %}
8
+ {%- if not date_string is defined %}
9
+ {%- set date_string = "26 Jul 2024" %}
10
+ {%- endif %}
11
+ {%- if not tools is defined %}
12
+ {%- set tools = none %}
13
+ {%- endif %}
14
+
15
+ {#- This block extracts the system message, so we can slot it into the right place. #}
16
+ {%- if messages[0]['role'] == 'system' %}
17
+ {%- set system_message = messages[0]['content']|trim %}
18
+ {%- set messages = messages[1:] %}
19
+ {%- else %}
20
+ {%- set system_message = "" %}
21
+ {%- endif %}
22
+
23
+ {#- System message + builtin tools #}
24
+ {{- "<|start_header_id|>system<|end_header_id|>\n\n" }}
25
+ {%- if builtin_tools is defined or tools is not none %}
26
+ {{- "Environment: ipython\n" }}
27
+ {%- endif %}
28
+ {%- if builtin_tools is defined %}
29
+ {{- "Tools: " + builtin_tools | reject('equalto', 'code_interpreter') | join(", ") + "\n\n"}}
30
+ {%- endif %}
31
+ {{- "Cutting Knowledge Date: December 2023\n" }}
32
+ {{- "Today Date: " + date_string + "\n\n" }}
33
+ {%- if tools is not none and not tools_in_user_message %}
34
+ {{- "You have access to the following functions. To call a function, please respond with JSON for a function call." }}
35
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
36
+ {{- "Do not use variables.\n\n" }}
37
+ {%- for t in tools %}
38
+ {{- t | tojson(indent=4) }}
39
+ {{- "\n\n" }}
40
+ {%- endfor %}
41
+ {%- endif %}
42
+ {{- system_message }}
43
+ {{- "<|eot_id|>" }}
44
+
45
+ {#- Custom tools are passed in a user message with some extra guidance #}
46
+ {%- if tools_in_user_message and not tools is none %}
47
+ {#- Extract the first user message so we can plug it in here #}
48
+ {%- if messages | length != 0 %}
49
+ {%- set first_user_message = messages[0]['content']|trim %}
50
+ {%- set messages = messages[1:] %}
51
+ {%- else %}
52
+ {{- raise_exception("Cannot put tools in the first user message when there's no first user message!") }}
53
+ {%- endif %}
54
+ {{- '<|start_header_id|>user<|end_header_id|>\n\n' -}}
55
+ {{- "Given the following functions, please respond with a JSON for a function call " }}
56
+ {{- "with its proper arguments that best answers the given prompt.\n\n" }}
57
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
58
+ {{- "Do not use variables.\n\n" }}
59
+ {%- for t in tools %}
60
+ {{- t | tojson(indent=4) }}
61
+ {{- "\n\n" }}
62
+ {%- endfor %}
63
+ {{- first_user_message + "<|eot_id|>"}}
64
+ {%- endif %}
65
+
66
+ {%- for message in messages %}
67
+ {%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}
68
+ {{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'+ message['content'] | trim + '<|eot_id|>' }}
69
+ {%- elif 'tool_calls' in message %}
70
+ {%- if not message.tool_calls|length == 1 %}
71
+ {{- raise_exception("This model only supports single tool-calls at once!") }}
72
+ {%- endif %}
73
+ {%- set tool_call = message.tool_calls[0].function %}
74
+ {%- if builtin_tools is defined and tool_call.name in builtin_tools %}
75
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' -}}
76
+ {{- "<|python_tag|>" + tool_call.name + ".call(" }}
77
+ {%- for arg_name, arg_val in tool_call.arguments | items %}
78
+ {{- arg_name + '="' + arg_val + '"' }}
79
+ {%- if not loop.last %}
80
+ {{- ", " }}
81
+ {%- endif %}
82
+ {%- endfor %}
83
+ {{- ")" }}
84
+ {%- else %}
85
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' -}}
86
+ {{- '{"name": "' + tool_call.name + '", ' }}
87
+ {{- '"parameters": ' }}
88
+ {{- tool_call.arguments | tojson }}
89
+ {{- "}" }}
90
+ {%- endif %}
91
+ {%- if builtin_tools is defined %}
92
+ {#- This means we're in ipython mode #}
93
+ {{- "<|eom_id|>" }}
94
+ {%- else %}
95
+ {{- "<|eot_id|>" }}
96
+ {%- endif %}
97
+ {%- elif message.role == "tool" or message.role == "ipython" %}
98
+ {{- "<|start_header_id|>ipython<|end_header_id|>\n\n" }}
99
+ {%- if message.content is mapping or message.content is iterable %}
100
+ {{- message.content | tojson }}
101
+ {%- else %}
102
+ {{- message.content }}
103
+ {%- endif %}
104
+ {{- "<|eot_id|>" }}
105
+ {%- endif %}
106
+ {%- endfor %}
107
+ {%- if add_generation_prompt %}
108
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' }}
109
+ {%- endif %}
empathy/checkpoint-939/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:379490fef5b9bf34f48733bc7918abffb9a3c4850e88d06eb7df9d048b190c1a
3
+ size 54745547
empathy/checkpoint-939/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:306e875b35c7ba49c246721674926c4364905aba32560771a2e499d5f56edc8f
3
+ size 14645
empathy/checkpoint-939/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a85b9d8d75b43f2deffbdf4c4ecce548860c65bc59c665b7c0dfc2918984f3a7
3
+ size 1465
empathy/checkpoint-939/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b9e4e7fb171f92fd137b777cc2714bf87d11576700a1dcd7a399e7bbe39537b
3
+ size 17209920
empathy/checkpoint-939/tokenizer_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "bos_token": "<|begin_of_text|>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "eos_token": "<|eot_id|>",
6
+ "is_local": false,
7
+ "model_input_names": [
8
+ "input_ids",
9
+ "attention_mask"
10
+ ],
11
+ "model_max_length": 131072,
12
+ "pad_token": "<|eot_id|>",
13
+ "tokenizer_class": "TokenizersBackend"
14
+ }
empathy/checkpoint-939/trainer_state.json ADDED
@@ -0,0 +1,964 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 3.0,
6
+ "eval_steps": 500,
7
+ "global_step": 939,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "entropy": 2.7027451276779173,
14
+ "epoch": 0.032,
15
+ "grad_norm": 0.263671875,
16
+ "learning_rate": 6.206896551724138e-05,
17
+ "loss": 2.8736827850341795,
18
+ "mean_token_accuracy": 0.45954256504774094,
19
+ "num_tokens": 55882.0,
20
+ "step": 10
21
+ },
22
+ {
23
+ "entropy": 2.2726758420467377,
24
+ "epoch": 0.064,
25
+ "grad_norm": 0.2431640625,
26
+ "learning_rate": 0.00013103448275862068,
27
+ "loss": 2.4437068939208983,
28
+ "mean_token_accuracy": 0.5048861749470234,
29
+ "num_tokens": 111677.0,
30
+ "step": 20
31
+ },
32
+ {
33
+ "entropy": 1.816487193107605,
34
+ "epoch": 0.096,
35
+ "grad_norm": 0.275390625,
36
+ "learning_rate": 0.0002,
37
+ "loss": 1.7309745788574218,
38
+ "mean_token_accuracy": 0.6087440773844719,
39
+ "num_tokens": 167580.0,
40
+ "step": 30
41
+ },
42
+ {
43
+ "entropy": 1.1870245933532715,
44
+ "epoch": 0.128,
45
+ "grad_norm": 0.275390625,
46
+ "learning_rate": 0.0001978021978021978,
47
+ "loss": 1.1468750953674316,
48
+ "mean_token_accuracy": 0.7256016343832016,
49
+ "num_tokens": 223679.0,
50
+ "step": 40
51
+ },
52
+ {
53
+ "entropy": 0.8109098121523857,
54
+ "epoch": 0.16,
55
+ "grad_norm": 0.333984375,
56
+ "learning_rate": 0.00019560439560439562,
57
+ "loss": 0.7429761409759521,
58
+ "mean_token_accuracy": 0.8220725610852242,
59
+ "num_tokens": 279556.0,
60
+ "step": 50
61
+ },
62
+ {
63
+ "entropy": 0.5473626285791398,
64
+ "epoch": 0.192,
65
+ "grad_norm": 0.267578125,
66
+ "learning_rate": 0.00019340659340659342,
67
+ "loss": 0.4774918556213379,
68
+ "mean_token_accuracy": 0.8853456184267998,
69
+ "num_tokens": 335848.0,
70
+ "step": 60
71
+ },
72
+ {
73
+ "entropy": 0.35489378273487093,
74
+ "epoch": 0.224,
75
+ "grad_norm": 0.263671875,
76
+ "learning_rate": 0.00019120879120879122,
77
+ "loss": 0.2937335968017578,
78
+ "mean_token_accuracy": 0.9316855490207672,
79
+ "num_tokens": 392355.0,
80
+ "step": 70
81
+ },
82
+ {
83
+ "entropy": 0.25775772109627726,
84
+ "epoch": 0.256,
85
+ "grad_norm": 0.251953125,
86
+ "learning_rate": 0.00018901098901098903,
87
+ "loss": 0.21065135002136232,
88
+ "mean_token_accuracy": 0.9512736469507217,
89
+ "num_tokens": 448792.0,
90
+ "step": 80
91
+ },
92
+ {
93
+ "entropy": 0.20272715911269187,
94
+ "epoch": 0.288,
95
+ "grad_norm": 0.1953125,
96
+ "learning_rate": 0.00018681318681318683,
97
+ "loss": 0.1630323886871338,
98
+ "mean_token_accuracy": 0.9601437523961067,
99
+ "num_tokens": 505118.0,
100
+ "step": 90
101
+ },
102
+ {
103
+ "entropy": 0.15652943775057793,
104
+ "epoch": 0.32,
105
+ "grad_norm": 0.14453125,
106
+ "learning_rate": 0.00018461538461538463,
107
+ "loss": 0.13783388137817382,
108
+ "mean_token_accuracy": 0.9648320719599723,
109
+ "num_tokens": 562043.0,
110
+ "step": 100
111
+ },
112
+ {
113
+ "entropy": 0.13883790336549282,
114
+ "epoch": 0.352,
115
+ "grad_norm": 0.1494140625,
116
+ "learning_rate": 0.0001824175824175824,
117
+ "loss": 0.1205636978149414,
118
+ "mean_token_accuracy": 0.9679180562496186,
119
+ "num_tokens": 618572.0,
120
+ "step": 110
121
+ },
122
+ {
123
+ "entropy": 0.1294231442734599,
124
+ "epoch": 0.384,
125
+ "grad_norm": 0.142578125,
126
+ "learning_rate": 0.00018021978021978024,
127
+ "loss": 0.10652012825012207,
128
+ "mean_token_accuracy": 0.9696741059422493,
129
+ "num_tokens": 674953.0,
130
+ "step": 120
131
+ },
132
+ {
133
+ "entropy": 0.12372843902558088,
134
+ "epoch": 0.416,
135
+ "grad_norm": 0.091796875,
136
+ "learning_rate": 0.00017802197802197802,
137
+ "loss": 0.09909504652023315,
138
+ "mean_token_accuracy": 0.9700024798512459,
139
+ "num_tokens": 730476.0,
140
+ "step": 130
141
+ },
142
+ {
143
+ "entropy": 0.11392448395490647,
144
+ "epoch": 0.448,
145
+ "grad_norm": 0.1376953125,
146
+ "learning_rate": 0.00017582417582417582,
147
+ "loss": 0.09517329335212707,
148
+ "mean_token_accuracy": 0.9697632819414139,
149
+ "num_tokens": 786626.0,
150
+ "step": 140
151
+ },
152
+ {
153
+ "entropy": 0.10535773131996393,
154
+ "epoch": 0.48,
155
+ "grad_norm": 0.12451171875,
156
+ "learning_rate": 0.00017362637362637365,
157
+ "loss": 0.08859028220176697,
158
+ "mean_token_accuracy": 0.9710599616169929,
159
+ "num_tokens": 842780.0,
160
+ "step": 150
161
+ },
162
+ {
163
+ "entropy": 0.1028441235423088,
164
+ "epoch": 0.512,
165
+ "grad_norm": 0.1455078125,
166
+ "learning_rate": 0.00017142857142857143,
167
+ "loss": 0.08795538544654846,
168
+ "mean_token_accuracy": 0.9719248920679092,
169
+ "num_tokens": 898828.0,
170
+ "step": 160
171
+ },
172
+ {
173
+ "entropy": 0.09524063598364592,
174
+ "epoch": 0.544,
175
+ "grad_norm": 0.17578125,
176
+ "learning_rate": 0.00016923076923076923,
177
+ "loss": 0.0860197126865387,
178
+ "mean_token_accuracy": 0.9718463212251663,
179
+ "num_tokens": 955109.0,
180
+ "step": 170
181
+ },
182
+ {
183
+ "entropy": 0.09921295028179884,
184
+ "epoch": 0.576,
185
+ "grad_norm": 0.11328125,
186
+ "learning_rate": 0.00016703296703296706,
187
+ "loss": 0.08638875484466553,
188
+ "mean_token_accuracy": 0.9711567848920822,
189
+ "num_tokens": 1010735.0,
190
+ "step": 180
191
+ },
192
+ {
193
+ "entropy": 0.09508467447012663,
194
+ "epoch": 0.608,
195
+ "grad_norm": 0.10888671875,
196
+ "learning_rate": 0.00016483516483516484,
197
+ "loss": 0.08325063586235046,
198
+ "mean_token_accuracy": 0.9720648273825645,
199
+ "num_tokens": 1067222.0,
200
+ "step": 190
201
+ },
202
+ {
203
+ "entropy": 0.09094681274145841,
204
+ "epoch": 0.64,
205
+ "grad_norm": 0.0927734375,
206
+ "learning_rate": 0.00016263736263736264,
207
+ "loss": 0.0826878547668457,
208
+ "mean_token_accuracy": 0.9720210924744606,
209
+ "num_tokens": 1122748.0,
210
+ "step": 200
211
+ },
212
+ {
213
+ "entropy": 0.09381346199661493,
214
+ "epoch": 0.672,
215
+ "grad_norm": 0.0634765625,
216
+ "learning_rate": 0.00016043956043956044,
217
+ "loss": 0.08256186246871948,
218
+ "mean_token_accuracy": 0.9720484882593154,
219
+ "num_tokens": 1178643.0,
220
+ "step": 210
221
+ },
222
+ {
223
+ "entropy": 0.09017684925347566,
224
+ "epoch": 0.704,
225
+ "grad_norm": 0.09375,
226
+ "learning_rate": 0.00015824175824175824,
227
+ "loss": 0.08275171518325805,
228
+ "mean_token_accuracy": 0.9719254687428475,
229
+ "num_tokens": 1234871.0,
230
+ "step": 220
231
+ },
232
+ {
233
+ "entropy": 0.08761313948780299,
234
+ "epoch": 0.736,
235
+ "grad_norm": 0.119140625,
236
+ "learning_rate": 0.00015604395604395605,
237
+ "loss": 0.08114268779754638,
238
+ "mean_token_accuracy": 0.9720645502209664,
239
+ "num_tokens": 1290368.0,
240
+ "step": 230
241
+ },
242
+ {
243
+ "entropy": 0.08477144911885262,
244
+ "epoch": 0.768,
245
+ "grad_norm": 0.08251953125,
246
+ "learning_rate": 0.00015384615384615385,
247
+ "loss": 0.07812036871910095,
248
+ "mean_token_accuracy": 0.9725761353969574,
249
+ "num_tokens": 1347259.0,
250
+ "step": 240
251
+ },
252
+ {
253
+ "entropy": 0.0840575411915779,
254
+ "epoch": 0.8,
255
+ "grad_norm": 0.10595703125,
256
+ "learning_rate": 0.00015164835164835165,
257
+ "loss": 0.07779947519302369,
258
+ "mean_token_accuracy": 0.9721481159329415,
259
+ "num_tokens": 1404092.0,
260
+ "step": 250
261
+ },
262
+ {
263
+ "entropy": 0.08560035470873117,
264
+ "epoch": 0.832,
265
+ "grad_norm": 0.0654296875,
266
+ "learning_rate": 0.00014945054945054946,
267
+ "loss": 0.07757753729820252,
268
+ "mean_token_accuracy": 0.9719480901956559,
269
+ "num_tokens": 1460432.0,
270
+ "step": 260
271
+ },
272
+ {
273
+ "entropy": 0.08481895942240954,
274
+ "epoch": 0.864,
275
+ "grad_norm": 0.07080078125,
276
+ "learning_rate": 0.00014725274725274726,
277
+ "loss": 0.07894231081008911,
278
+ "mean_token_accuracy": 0.9720895797014236,
279
+ "num_tokens": 1516153.0,
280
+ "step": 270
281
+ },
282
+ {
283
+ "entropy": 0.08592585939913988,
284
+ "epoch": 0.896,
285
+ "grad_norm": 0.08251953125,
286
+ "learning_rate": 0.00014505494505494506,
287
+ "loss": 0.07765401601791382,
288
+ "mean_token_accuracy": 0.973065185546875,
289
+ "num_tokens": 1572583.0,
290
+ "step": 280
291
+ },
292
+ {
293
+ "entropy": 0.0848662929609418,
294
+ "epoch": 0.928,
295
+ "grad_norm": 0.1103515625,
296
+ "learning_rate": 0.00014285714285714287,
297
+ "loss": 0.07779905796051026,
298
+ "mean_token_accuracy": 0.9732325404882431,
299
+ "num_tokens": 1628569.0,
300
+ "step": 290
301
+ },
302
+ {
303
+ "entropy": 0.08204772900789976,
304
+ "epoch": 0.96,
305
+ "grad_norm": 0.059326171875,
306
+ "learning_rate": 0.00014065934065934067,
307
+ "loss": 0.07686090469360352,
308
+ "mean_token_accuracy": 0.9732487082481385,
309
+ "num_tokens": 1684838.0,
310
+ "step": 300
311
+ },
312
+ {
313
+ "entropy": 0.08476632833480835,
314
+ "epoch": 0.992,
315
+ "grad_norm": 0.10791015625,
316
+ "learning_rate": 0.00013846153846153847,
317
+ "loss": 0.076432603597641,
318
+ "mean_token_accuracy": 0.9719811454415321,
319
+ "num_tokens": 1741902.0,
320
+ "step": 310
321
+ },
322
+ {
323
+ "entropy": 0.07945429298438524,
324
+ "epoch": 1.0224,
325
+ "grad_norm": 0.0634765625,
326
+ "learning_rate": 0.00013626373626373628,
327
+ "loss": 0.07495068311691284,
328
+ "mean_token_accuracy": 0.9731942399551994,
329
+ "num_tokens": 1795784.0,
330
+ "step": 320
331
+ },
332
+ {
333
+ "entropy": 0.0806297117844224,
334
+ "epoch": 1.0544,
335
+ "grad_norm": 0.076171875,
336
+ "learning_rate": 0.00013406593406593405,
337
+ "loss": 0.07468733191490173,
338
+ "mean_token_accuracy": 0.9737721592187881,
339
+ "num_tokens": 1852266.0,
340
+ "step": 330
341
+ },
342
+ {
343
+ "entropy": 0.07759423423558473,
344
+ "epoch": 1.0864,
345
+ "grad_norm": 0.07763671875,
346
+ "learning_rate": 0.00013186813186813188,
347
+ "loss": 0.07366250157356262,
348
+ "mean_token_accuracy": 0.9733460173010826,
349
+ "num_tokens": 1908148.0,
350
+ "step": 340
351
+ },
352
+ {
353
+ "entropy": 0.08119724169373513,
354
+ "epoch": 1.1184,
355
+ "grad_norm": 0.050537109375,
356
+ "learning_rate": 0.0001296703296703297,
357
+ "loss": 0.07312369346618652,
358
+ "mean_token_accuracy": 0.97357087880373,
359
+ "num_tokens": 1964791.0,
360
+ "step": 350
361
+ },
362
+ {
363
+ "entropy": 0.07786643356084824,
364
+ "epoch": 1.1504,
365
+ "grad_norm": 0.04833984375,
366
+ "learning_rate": 0.00012747252747252746,
367
+ "loss": 0.07509608864784241,
368
+ "mean_token_accuracy": 0.9725616052746773,
369
+ "num_tokens": 2020416.0,
370
+ "step": 360
371
+ },
372
+ {
373
+ "entropy": 0.07941898722201586,
374
+ "epoch": 1.1824,
375
+ "grad_norm": 0.068359375,
376
+ "learning_rate": 0.00012527472527472527,
377
+ "loss": 0.0736158847808838,
378
+ "mean_token_accuracy": 0.9736046925187111,
379
+ "num_tokens": 2076384.0,
380
+ "step": 370
381
+ },
382
+ {
383
+ "entropy": 0.0761857332661748,
384
+ "epoch": 1.2144,
385
+ "grad_norm": 0.1337890625,
386
+ "learning_rate": 0.0001230769230769231,
387
+ "loss": 0.07236781120300292,
388
+ "mean_token_accuracy": 0.9739944905042648,
389
+ "num_tokens": 2132795.0,
390
+ "step": 380
391
+ },
392
+ {
393
+ "entropy": 0.07662795521318913,
394
+ "epoch": 1.2464,
395
+ "grad_norm": 0.1474609375,
396
+ "learning_rate": 0.00012087912087912087,
397
+ "loss": 0.07198388576507568,
398
+ "mean_token_accuracy": 0.9738625854253768,
399
+ "num_tokens": 2189096.0,
400
+ "step": 390
401
+ },
402
+ {
403
+ "entropy": 0.07697693575173617,
404
+ "epoch": 1.2784,
405
+ "grad_norm": 0.0634765625,
406
+ "learning_rate": 0.00011868131868131869,
407
+ "loss": 0.07262890338897705,
408
+ "mean_token_accuracy": 0.9728588461875916,
409
+ "num_tokens": 2245179.0,
410
+ "step": 400
411
+ },
412
+ {
413
+ "entropy": 0.07774264849722386,
414
+ "epoch": 1.3104,
415
+ "grad_norm": 0.062255859375,
416
+ "learning_rate": 0.0001164835164835165,
417
+ "loss": 0.07280476689338684,
418
+ "mean_token_accuracy": 0.9733426690101623,
419
+ "num_tokens": 2300662.0,
420
+ "step": 410
421
+ },
422
+ {
423
+ "entropy": 0.07713319007307291,
424
+ "epoch": 1.3424,
425
+ "grad_norm": 0.048583984375,
426
+ "learning_rate": 0.00011428571428571428,
427
+ "loss": 0.07291589379310608,
428
+ "mean_token_accuracy": 0.9731875821948052,
429
+ "num_tokens": 2356598.0,
430
+ "step": 420
431
+ },
432
+ {
433
+ "entropy": 0.07562742829322815,
434
+ "epoch": 1.3744,
435
+ "grad_norm": 0.060791015625,
436
+ "learning_rate": 0.0001120879120879121,
437
+ "loss": 0.0718300998210907,
438
+ "mean_token_accuracy": 0.9734191030263901,
439
+ "num_tokens": 2412737.0,
440
+ "step": 430
441
+ },
442
+ {
443
+ "entropy": 0.07663317285478115,
444
+ "epoch": 1.4064,
445
+ "grad_norm": 0.058349609375,
446
+ "learning_rate": 0.0001098901098901099,
447
+ "loss": 0.0707779049873352,
448
+ "mean_token_accuracy": 0.9742564350366593,
449
+ "num_tokens": 2469165.0,
450
+ "step": 440
451
+ },
452
+ {
453
+ "entropy": 0.07312421463429927,
454
+ "epoch": 1.4384000000000001,
455
+ "grad_norm": 0.0654296875,
456
+ "learning_rate": 0.0001076923076923077,
457
+ "loss": 0.07058998346328735,
458
+ "mean_token_accuracy": 0.9742231473326683,
459
+ "num_tokens": 2525741.0,
460
+ "step": 450
461
+ },
462
+ {
463
+ "entropy": 0.07522066552191972,
464
+ "epoch": 1.4704,
465
+ "grad_norm": 0.052978515625,
466
+ "learning_rate": 0.0001054945054945055,
467
+ "loss": 0.07139057517051697,
468
+ "mean_token_accuracy": 0.9742771983146667,
469
+ "num_tokens": 2582350.0,
470
+ "step": 460
471
+ },
472
+ {
473
+ "entropy": 0.07417810596525669,
474
+ "epoch": 1.5024,
475
+ "grad_norm": 0.04443359375,
476
+ "learning_rate": 0.00010329670329670331,
477
+ "loss": 0.0715557336807251,
478
+ "mean_token_accuracy": 0.9731055706739425,
479
+ "num_tokens": 2638763.0,
480
+ "step": 470
481
+ },
482
+ {
483
+ "entropy": 0.07531943805515766,
484
+ "epoch": 1.5344,
485
+ "grad_norm": 0.08349609375,
486
+ "learning_rate": 0.0001010989010989011,
487
+ "loss": 0.07040765285491943,
488
+ "mean_token_accuracy": 0.9743428856134415,
489
+ "num_tokens": 2695023.0,
490
+ "step": 480
491
+ },
492
+ {
493
+ "entropy": 0.0734874278306961,
494
+ "epoch": 1.5664,
495
+ "grad_norm": 0.06591796875,
496
+ "learning_rate": 9.89010989010989e-05,
497
+ "loss": 0.07134199142456055,
498
+ "mean_token_accuracy": 0.9733200684189797,
499
+ "num_tokens": 2751169.0,
500
+ "step": 490
501
+ },
502
+ {
503
+ "entropy": 0.07535444647073745,
504
+ "epoch": 1.5984,
505
+ "grad_norm": 0.05615234375,
506
+ "learning_rate": 9.670329670329671e-05,
507
+ "loss": 0.07132225036621094,
508
+ "mean_token_accuracy": 0.9734165251255036,
509
+ "num_tokens": 2807473.0,
510
+ "step": 500
511
+ },
512
+ {
513
+ "entropy": 0.07486743628978729,
514
+ "epoch": 1.6303999999999998,
515
+ "grad_norm": 0.04736328125,
516
+ "learning_rate": 9.450549450549451e-05,
517
+ "loss": 0.0716746211051941,
518
+ "mean_token_accuracy": 0.9733885332942009,
519
+ "num_tokens": 2863590.0,
520
+ "step": 510
521
+ },
522
+ {
523
+ "entropy": 0.07372102495282888,
524
+ "epoch": 1.6623999999999999,
525
+ "grad_norm": 0.0654296875,
526
+ "learning_rate": 9.230769230769232e-05,
527
+ "loss": 0.0713922917842865,
528
+ "mean_token_accuracy": 0.973819674551487,
529
+ "num_tokens": 2919811.0,
530
+ "step": 520
531
+ },
532
+ {
533
+ "entropy": 0.07463319394737482,
534
+ "epoch": 1.6944,
535
+ "grad_norm": 0.0419921875,
536
+ "learning_rate": 9.010989010989012e-05,
537
+ "loss": 0.0703410565853119,
538
+ "mean_token_accuracy": 0.9737539097666741,
539
+ "num_tokens": 2976301.0,
540
+ "step": 530
541
+ },
542
+ {
543
+ "entropy": 0.0717978198081255,
544
+ "epoch": 1.7264,
545
+ "grad_norm": 0.052001953125,
546
+ "learning_rate": 8.791208791208791e-05,
547
+ "loss": 0.06898372173309326,
548
+ "mean_token_accuracy": 0.974400269985199,
549
+ "num_tokens": 3032771.0,
550
+ "step": 540
551
+ },
552
+ {
553
+ "entropy": 0.07407103907316923,
554
+ "epoch": 1.7584,
555
+ "grad_norm": 0.05517578125,
556
+ "learning_rate": 8.571428571428571e-05,
557
+ "loss": 0.07060829997062683,
558
+ "mean_token_accuracy": 0.9738653019070626,
559
+ "num_tokens": 3088405.0,
560
+ "step": 550
561
+ },
562
+ {
563
+ "entropy": 0.07348943576216697,
564
+ "epoch": 1.7904,
565
+ "grad_norm": 0.052490234375,
566
+ "learning_rate": 8.351648351648353e-05,
567
+ "loss": 0.06981388926506042,
568
+ "mean_token_accuracy": 0.973979789018631,
569
+ "num_tokens": 3144186.0,
570
+ "step": 560
571
+ },
572
+ {
573
+ "entropy": 0.07443872913718223,
574
+ "epoch": 1.8224,
575
+ "grad_norm": 0.04638671875,
576
+ "learning_rate": 8.131868131868132e-05,
577
+ "loss": 0.07104510068893433,
578
+ "mean_token_accuracy": 0.97396722137928,
579
+ "num_tokens": 3200274.0,
580
+ "step": 570
581
+ },
582
+ {
583
+ "entropy": 0.07258394770324231,
584
+ "epoch": 1.8544,
585
+ "grad_norm": 0.0673828125,
586
+ "learning_rate": 7.912087912087912e-05,
587
+ "loss": 0.06858785152435302,
588
+ "mean_token_accuracy": 0.9745583027601242,
589
+ "num_tokens": 3257100.0,
590
+ "step": 580
591
+ },
592
+ {
593
+ "entropy": 0.07335705403238535,
594
+ "epoch": 1.8864,
595
+ "grad_norm": 0.04296875,
596
+ "learning_rate": 7.692307692307693e-05,
597
+ "loss": 0.0689892292022705,
598
+ "mean_token_accuracy": 0.9740391105413437,
599
+ "num_tokens": 3313764.0,
600
+ "step": 590
601
+ },
602
+ {
603
+ "entropy": 0.0722169006243348,
604
+ "epoch": 1.9184,
605
+ "grad_norm": 0.052734375,
606
+ "learning_rate": 7.472527472527473e-05,
607
+ "loss": 0.06918958425521851,
608
+ "mean_token_accuracy": 0.9745689764618873,
609
+ "num_tokens": 3370053.0,
610
+ "step": 600
611
+ },
612
+ {
613
+ "entropy": 0.0719969030469656,
614
+ "epoch": 1.9504000000000001,
615
+ "grad_norm": 0.05078125,
616
+ "learning_rate": 7.252747252747253e-05,
617
+ "loss": 0.07059705257415771,
618
+ "mean_token_accuracy": 0.9735003352165222,
619
+ "num_tokens": 3425403.0,
620
+ "step": 610
621
+ },
622
+ {
623
+ "entropy": 0.07328992187976838,
624
+ "epoch": 1.9824000000000002,
625
+ "grad_norm": 0.05029296875,
626
+ "learning_rate": 7.032967032967034e-05,
627
+ "loss": 0.06893991827964782,
628
+ "mean_token_accuracy": 0.974341906607151,
629
+ "num_tokens": 3481888.0,
630
+ "step": 620
631
+ },
632
+ {
633
+ "entropy": 0.07355283553663053,
634
+ "epoch": 2.0128,
635
+ "grad_norm": 0.05419921875,
636
+ "learning_rate": 6.813186813186814e-05,
637
+ "loss": 0.06852576732635499,
638
+ "mean_token_accuracy": 0.9741844628986559,
639
+ "num_tokens": 3535234.0,
640
+ "step": 630
641
+ },
642
+ {
643
+ "entropy": 0.07177970744669437,
644
+ "epoch": 2.0448,
645
+ "grad_norm": 0.0458984375,
646
+ "learning_rate": 6.593406593406594e-05,
647
+ "loss": 0.06823881268501282,
648
+ "mean_token_accuracy": 0.9741811975836754,
649
+ "num_tokens": 3590735.0,
650
+ "step": 640
651
+ },
652
+ {
653
+ "entropy": 0.07052574083209037,
654
+ "epoch": 2.0768,
655
+ "grad_norm": 0.047119140625,
656
+ "learning_rate": 6.373626373626373e-05,
657
+ "loss": 0.06783221364021301,
658
+ "mean_token_accuracy": 0.9742385044693946,
659
+ "num_tokens": 3647317.0,
660
+ "step": 650
661
+ },
662
+ {
663
+ "entropy": 0.07118275295943022,
664
+ "epoch": 2.1088,
665
+ "grad_norm": 0.057373046875,
666
+ "learning_rate": 6.153846153846155e-05,
667
+ "loss": 0.06789056658744812,
668
+ "mean_token_accuracy": 0.9747925192117691,
669
+ "num_tokens": 3704190.0,
670
+ "step": 660
671
+ },
672
+ {
673
+ "entropy": 0.07141004446893931,
674
+ "epoch": 2.1408,
675
+ "grad_norm": 0.046630859375,
676
+ "learning_rate": 5.9340659340659345e-05,
677
+ "loss": 0.06710809469223022,
678
+ "mean_token_accuracy": 0.9745214998722076,
679
+ "num_tokens": 3760496.0,
680
+ "step": 670
681
+ },
682
+ {
683
+ "entropy": 0.06936982274055481,
684
+ "epoch": 2.1728,
685
+ "grad_norm": 0.052490234375,
686
+ "learning_rate": 5.714285714285714e-05,
687
+ "loss": 0.06704961657524108,
688
+ "mean_token_accuracy": 0.9749385923147201,
689
+ "num_tokens": 3817106.0,
690
+ "step": 680
691
+ },
692
+ {
693
+ "entropy": 0.0714575270190835,
694
+ "epoch": 2.2048,
695
+ "grad_norm": 0.04833984375,
696
+ "learning_rate": 5.494505494505495e-05,
697
+ "loss": 0.06737448573112488,
698
+ "mean_token_accuracy": 0.9740116745233536,
699
+ "num_tokens": 3873694.0,
700
+ "step": 690
701
+ },
702
+ {
703
+ "entropy": 0.07101229522377253,
704
+ "epoch": 2.2368,
705
+ "grad_norm": 0.05517578125,
706
+ "learning_rate": 5.274725274725275e-05,
707
+ "loss": 0.0672797441482544,
708
+ "mean_token_accuracy": 0.974830585718155,
709
+ "num_tokens": 3929897.0,
710
+ "step": 700
711
+ },
712
+ {
713
+ "entropy": 0.0700510073453188,
714
+ "epoch": 2.2688,
715
+ "grad_norm": 0.052734375,
716
+ "learning_rate": 5.054945054945055e-05,
717
+ "loss": 0.06735477447509766,
718
+ "mean_token_accuracy": 0.9748806655406952,
719
+ "num_tokens": 3985906.0,
720
+ "step": 710
721
+ },
722
+ {
723
+ "entropy": 0.07011591829359531,
724
+ "epoch": 2.3008,
725
+ "grad_norm": 0.0478515625,
726
+ "learning_rate": 4.8351648351648355e-05,
727
+ "loss": 0.06764240264892578,
728
+ "mean_token_accuracy": 0.9744778007268906,
729
+ "num_tokens": 4042082.0,
730
+ "step": 720
731
+ },
732
+ {
733
+ "entropy": 0.07075654622167349,
734
+ "epoch": 2.3327999999999998,
735
+ "grad_norm": 0.04345703125,
736
+ "learning_rate": 4.615384615384616e-05,
737
+ "loss": 0.0661679446697235,
738
+ "mean_token_accuracy": 0.9753401562571525,
739
+ "num_tokens": 4098909.0,
740
+ "step": 730
741
+ },
742
+ {
743
+ "entropy": 0.07101095058023929,
744
+ "epoch": 2.3648,
745
+ "grad_norm": 0.0498046875,
746
+ "learning_rate": 4.3956043956043955e-05,
747
+ "loss": 0.06830476522445679,
748
+ "mean_token_accuracy": 0.9741760551929474,
749
+ "num_tokens": 4154222.0,
750
+ "step": 740
751
+ },
752
+ {
753
+ "entropy": 0.07036219611763954,
754
+ "epoch": 2.3968,
755
+ "grad_norm": 0.046875,
756
+ "learning_rate": 4.1758241758241765e-05,
757
+ "loss": 0.06679952144622803,
758
+ "mean_token_accuracy": 0.9747874498367309,
759
+ "num_tokens": 4209907.0,
760
+ "step": 750
761
+ },
762
+ {
763
+ "entropy": 0.06965819718316198,
764
+ "epoch": 2.4288,
765
+ "grad_norm": 0.0478515625,
766
+ "learning_rate": 3.956043956043956e-05,
767
+ "loss": 0.06658568978309631,
768
+ "mean_token_accuracy": 0.9753396198153496,
769
+ "num_tokens": 4266708.0,
770
+ "step": 760
771
+ },
772
+ {
773
+ "entropy": 0.07088542385026812,
774
+ "epoch": 2.4608,
775
+ "grad_norm": 0.047607421875,
776
+ "learning_rate": 3.7362637362637365e-05,
777
+ "loss": 0.06808845400810241,
778
+ "mean_token_accuracy": 0.9744594439864158,
779
+ "num_tokens": 4322203.0,
780
+ "step": 770
781
+ },
782
+ {
783
+ "entropy": 0.07002440262585878,
784
+ "epoch": 2.4928,
785
+ "grad_norm": 0.04931640625,
786
+ "learning_rate": 3.516483516483517e-05,
787
+ "loss": 0.06674610376358033,
788
+ "mean_token_accuracy": 0.9749701410531998,
789
+ "num_tokens": 4378286.0,
790
+ "step": 780
791
+ },
792
+ {
793
+ "entropy": 0.07156520402058959,
794
+ "epoch": 2.5248,
795
+ "grad_norm": 0.07763671875,
796
+ "learning_rate": 3.296703296703297e-05,
797
+ "loss": 0.06783879399299622,
798
+ "mean_token_accuracy": 0.9742916822433472,
799
+ "num_tokens": 4433890.0,
800
+ "step": 790
801
+ },
802
+ {
803
+ "entropy": 0.07110334262251854,
804
+ "epoch": 2.5568,
805
+ "grad_norm": 0.05078125,
806
+ "learning_rate": 3.0769230769230774e-05,
807
+ "loss": 0.06623314619064331,
808
+ "mean_token_accuracy": 0.9757025733590126,
809
+ "num_tokens": 4490228.0,
810
+ "step": 800
811
+ },
812
+ {
813
+ "entropy": 0.07086261715739965,
814
+ "epoch": 2.5888,
815
+ "grad_norm": 0.053466796875,
816
+ "learning_rate": 2.857142857142857e-05,
817
+ "loss": 0.06686291694641114,
818
+ "mean_token_accuracy": 0.974934495985508,
819
+ "num_tokens": 4546293.0,
820
+ "step": 810
821
+ },
822
+ {
823
+ "entropy": 0.06940543595701457,
824
+ "epoch": 2.6208,
825
+ "grad_norm": 0.048583984375,
826
+ "learning_rate": 2.6373626373626374e-05,
827
+ "loss": 0.06693726181983947,
828
+ "mean_token_accuracy": 0.9744124352931977,
829
+ "num_tokens": 4602398.0,
830
+ "step": 820
831
+ },
832
+ {
833
+ "entropy": 0.07065027058124543,
834
+ "epoch": 2.6528,
835
+ "grad_norm": 0.048095703125,
836
+ "learning_rate": 2.4175824175824177e-05,
837
+ "loss": 0.06692788600921631,
838
+ "mean_token_accuracy": 0.9747064486145973,
839
+ "num_tokens": 4658249.0,
840
+ "step": 830
841
+ },
842
+ {
843
+ "entropy": 0.0688671569339931,
844
+ "epoch": 2.6848,
845
+ "grad_norm": 0.051513671875,
846
+ "learning_rate": 2.1978021978021977e-05,
847
+ "loss": 0.06544087529182434,
848
+ "mean_token_accuracy": 0.9749023050069809,
849
+ "num_tokens": 4714387.0,
850
+ "step": 840
851
+ },
852
+ {
853
+ "entropy": 0.06899926522746683,
854
+ "epoch": 2.7168,
855
+ "grad_norm": 0.05517578125,
856
+ "learning_rate": 1.978021978021978e-05,
857
+ "loss": 0.06681140065193177,
858
+ "mean_token_accuracy": 0.975035460293293,
859
+ "num_tokens": 4770700.0,
860
+ "step": 850
861
+ },
862
+ {
863
+ "entropy": 0.06984850689768791,
864
+ "epoch": 2.7488,
865
+ "grad_norm": 0.058837890625,
866
+ "learning_rate": 1.7582417582417584e-05,
867
+ "loss": 0.06549346446990967,
868
+ "mean_token_accuracy": 0.9752413630485535,
869
+ "num_tokens": 4827270.0,
870
+ "step": 860
871
+ },
872
+ {
873
+ "entropy": 0.07006102437153458,
874
+ "epoch": 2.7808,
875
+ "grad_norm": 0.049072265625,
876
+ "learning_rate": 1.5384615384615387e-05,
877
+ "loss": 0.06601048707962036,
878
+ "mean_token_accuracy": 0.9751122385263443,
879
+ "num_tokens": 4883287.0,
880
+ "step": 870
881
+ },
882
+ {
883
+ "entropy": 0.07032166775316,
884
+ "epoch": 2.8128,
885
+ "grad_norm": 0.052001953125,
886
+ "learning_rate": 1.3186813186813187e-05,
887
+ "loss": 0.06602610349655151,
888
+ "mean_token_accuracy": 0.9746679022908211,
889
+ "num_tokens": 4939190.0,
890
+ "step": 880
891
+ },
892
+ {
893
+ "entropy": 0.06937383627519011,
894
+ "epoch": 2.8448,
895
+ "grad_norm": 0.05126953125,
896
+ "learning_rate": 1.0989010989010989e-05,
897
+ "loss": 0.06518557667732239,
898
+ "mean_token_accuracy": 0.9755709722638131,
899
+ "num_tokens": 4995747.0,
900
+ "step": 890
901
+ },
902
+ {
903
+ "entropy": 0.07017751513049006,
904
+ "epoch": 2.8768000000000002,
905
+ "grad_norm": 0.05126953125,
906
+ "learning_rate": 8.791208791208792e-06,
907
+ "loss": 0.06582036018371581,
908
+ "mean_token_accuracy": 0.9752513289451599,
909
+ "num_tokens": 5052336.0,
910
+ "step": 900
911
+ },
912
+ {
913
+ "entropy": 0.06967417653650046,
914
+ "epoch": 2.9088000000000003,
915
+ "grad_norm": 0.05419921875,
916
+ "learning_rate": 6.5934065934065935e-06,
917
+ "loss": 0.06589003205299378,
918
+ "mean_token_accuracy": 0.9749706491827965,
919
+ "num_tokens": 5108592.0,
920
+ "step": 910
921
+ },
922
+ {
923
+ "entropy": 0.06899928543716669,
924
+ "epoch": 2.9408,
925
+ "grad_norm": 0.05029296875,
926
+ "learning_rate": 4.395604395604396e-06,
927
+ "loss": 0.06553114652633667,
928
+ "mean_token_accuracy": 0.9755852162837982,
929
+ "num_tokens": 5165398.0,
930
+ "step": 920
931
+ },
932
+ {
933
+ "entropy": 0.06951060444116593,
934
+ "epoch": 2.9728,
935
+ "grad_norm": 0.055419921875,
936
+ "learning_rate": 2.197802197802198e-06,
937
+ "loss": 0.06614152193069459,
938
+ "mean_token_accuracy": 0.9750266820192337,
939
+ "num_tokens": 5221349.0,
940
+ "step": 930
941
+ }
942
+ ],
943
+ "logging_steps": 10,
944
+ "max_steps": 939,
945
+ "num_input_tokens_seen": 0,
946
+ "num_train_epochs": 3,
947
+ "save_steps": 500,
948
+ "stateful_callbacks": {
949
+ "TrainerControl": {
950
+ "args": {
951
+ "should_epoch_stop": false,
952
+ "should_evaluate": false,
953
+ "should_log": false,
954
+ "should_save": true,
955
+ "should_training_stop": true
956
+ },
957
+ "attributes": {}
958
+ }
959
+ },
960
+ "total_flos": 2.4543986745581568e+17,
961
+ "train_batch_size": 2,
962
+ "trial_name": null,
963
+ "trial_params": null
964
+ }
empathy/checkpoint-939/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e64d4e636d5c0eb6e65352529689dee60289c387b4536f3e0b17f48b352ee585
3
+ size 5585
empathy/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b9e4e7fb171f92fd137b777cc2714bf87d11576700a1dcd7a399e7bbe39537b
3
+ size 17209920
empathy/tokenizer_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "bos_token": "<|begin_of_text|>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "eos_token": "<|eot_id|>",
6
+ "is_local": false,
7
+ "model_input_names": [
8
+ "input_ids",
9
+ "attention_mask"
10
+ ],
11
+ "model_max_length": 131072,
12
+ "pad_token": "<|eot_id|>",
13
+ "tokenizer_class": "TokenizersBackend"
14
+ }