sea-rod commited on
Commit
9e1165f
·
verified ·
1 Parent(s): 39dab32

Upload folder using huggingface_hub

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +44 -0
  2. v5/DPO/DPO_10k/DPO_10k/README.md +209 -0
  3. v5/DPO/DPO_10k/DPO_10k/adapter_config.json +46 -0
  4. v5/DPO/DPO_10k/DPO_10k/adapter_model.safetensors +3 -0
  5. v5/DPO/DPO_10k/MDPO_10k/chat_template.jinja +93 -0
  6. v5/DPO/DPO_10k/MDPO_10k/config.json +40 -0
  7. v5/DPO/DPO_10k/MDPO_10k/generation_config.json +12 -0
  8. v5/DPO/DPO_10k/MDPO_10k/model.safetensors +3 -0
  9. v5/DPO/DPO_10k/MDPO_10k/tokenizer.json +3 -0
  10. v5/DPO/DPO_10k/MDPO_10k/tokenizer_config.json +14 -0
  11. v5/DPO/DPO_10k/lora/README.md +69 -0
  12. v5/DPO/DPO_10k/lora/checkpoint-1300/README.md +209 -0
  13. v5/DPO/DPO_10k/lora/checkpoint-1300/adapter_config.json +46 -0
  14. v5/DPO/DPO_10k/lora/checkpoint-1300/adapter_model.safetensors +3 -0
  15. v5/DPO/DPO_10k/lora/checkpoint-1300/chat_template.jinja +93 -0
  16. v5/DPO/DPO_10k/lora/checkpoint-1300/optimizer.pt +3 -0
  17. v5/DPO/DPO_10k/lora/checkpoint-1300/rng_state.pth +3 -0
  18. v5/DPO/DPO_10k/lora/checkpoint-1300/scaler.pt +3 -0
  19. v5/DPO/DPO_10k/lora/checkpoint-1300/scheduler.pt +3 -0
  20. v5/DPO/DPO_10k/lora/checkpoint-1300/tokenizer.json +3 -0
  21. v5/DPO/DPO_10k/lora/checkpoint-1300/tokenizer_config.json +14 -0
  22. v5/DPO/DPO_10k/lora/checkpoint-1300/trainer_state.json +2192 -0
  23. v5/DPO/DPO_10k/lora/checkpoint-1300/training_args.bin +3 -0
  24. v5/DPO/DPO_10k/lora/checkpoint-2400/README.md +209 -0
  25. v5/DPO/DPO_10k/lora/checkpoint-2400/adapter_config.json +46 -0
  26. v5/DPO/DPO_10k/lora/checkpoint-2400/adapter_model.safetensors +3 -0
  27. v5/DPO/DPO_10k/lora/checkpoint-2400/chat_template.jinja +93 -0
  28. v5/DPO/DPO_10k/lora/checkpoint-2400/optimizer.pt +3 -0
  29. v5/DPO/DPO_10k/lora/checkpoint-2400/rng_state.pth +3 -0
  30. v5/DPO/DPO_10k/lora/checkpoint-2400/scaler.pt +3 -0
  31. v5/DPO/DPO_10k/lora/checkpoint-2400/scheduler.pt +3 -0
  32. v5/DPO/DPO_10k/lora/checkpoint-2400/tokenizer.json +3 -0
  33. v5/DPO/DPO_10k/lora/checkpoint-2400/tokenizer_config.json +14 -0
  34. v5/DPO/DPO_10k/lora/checkpoint-2400/trainer_state.json +0 -0
  35. v5/DPO/DPO_10k/lora/checkpoint-2400/training_args.bin +3 -0
  36. v5/DPO/DPO_10k/lora/checkpoint-2500/README.md +209 -0
  37. v5/DPO/DPO_10k/lora/checkpoint-2500/adapter_config.json +46 -0
  38. v5/DPO/DPO_10k/lora/checkpoint-2500/adapter_model.safetensors +3 -0
  39. v5/DPO/DPO_10k/lora/checkpoint-2500/chat_template.jinja +93 -0
  40. v5/DPO/DPO_10k/lora/checkpoint-2500/optimizer.pt +3 -0
  41. v5/DPO/DPO_10k/lora/checkpoint-2500/rng_state.pth +3 -0
  42. v5/DPO/DPO_10k/lora/checkpoint-2500/scaler.pt +3 -0
  43. v5/DPO/DPO_10k/lora/checkpoint-2500/scheduler.pt +3 -0
  44. v5/DPO/DPO_10k/lora/checkpoint-2500/tokenizer.json +3 -0
  45. v5/DPO/DPO_10k/lora/checkpoint-2500/tokenizer_config.json +14 -0
  46. v5/DPO/DPO_10k/lora/checkpoint-2500/trainer_state.json +0 -0
  47. v5/DPO/DPO_10k/lora/checkpoint-2500/training_args.bin +3 -0
  48. v5/DPO/DPO_1k/DPO_1k/README.md +209 -0
  49. v5/DPO/DPO_1k/DPO_1k/adapter_config.json +46 -0
  50. v5/DPO/DPO_1k/DPO_1k/adapter_model.safetensors +3 -0
.gitattributes CHANGED
@@ -78,3 +78,47 @@ v4/DPO/DPO_5k/MDPO_5k/tokenizer.json filter=lfs diff=lfs merge=lfs -text
78
  v4/DPO/DPO_5k/lora/checkpoint-1000/tokenizer.json filter=lfs diff=lfs merge=lfs -text
79
  v4/DPO/DPO_5k/lora/checkpoint-1250/tokenizer.json filter=lfs diff=lfs merge=lfs -text
80
  v4/DPO/DPO_5k/lora/checkpoint-500/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
78
  v4/DPO/DPO_5k/lora/checkpoint-1000/tokenizer.json filter=lfs diff=lfs merge=lfs -text
79
  v4/DPO/DPO_5k/lora/checkpoint-1250/tokenizer.json filter=lfs diff=lfs merge=lfs -text
80
  v4/DPO/DPO_5k/lora/checkpoint-500/tokenizer.json filter=lfs diff=lfs merge=lfs -text
81
+ v5/DPO/DPO_10k/MDPO_10k/tokenizer.json filter=lfs diff=lfs merge=lfs -text
82
+ v5/DPO/DPO_10k/lora/checkpoint-1300/tokenizer.json filter=lfs diff=lfs merge=lfs -text
83
+ v5/DPO/DPO_10k/lora/checkpoint-2400/tokenizer.json filter=lfs diff=lfs merge=lfs -text
84
+ v5/DPO/DPO_10k/lora/checkpoint-2500/tokenizer.json filter=lfs diff=lfs merge=lfs -text
85
+ v5/DPO/DPO_1k/MDPO_1k/tokenizer.json filter=lfs diff=lfs merge=lfs -text
86
+ v5/DPO/DPO_1k/lora/checkpoint-240/tokenizer.json filter=lfs diff=lfs merge=lfs -text
87
+ v5/DPO/DPO_1k/lora/checkpoint-250/tokenizer.json filter=lfs diff=lfs merge=lfs -text
88
+ v5/DPO/DPO_1k/lora/checkpoint-60/tokenizer.json filter=lfs diff=lfs merge=lfs -text
89
+ v5/DPO/DPO_5k/MDPO_5k/tokenizer.json filter=lfs diff=lfs merge=lfs -text
90
+ v5/DPO/DPO_5k/lora/checkpoint-1200/tokenizer.json filter=lfs diff=lfs merge=lfs -text
91
+ v5/DPO/DPO_5k/lora/checkpoint-1250/tokenizer.json filter=lfs diff=lfs merge=lfs -text
92
+ v5/DPO/DPO_5k/lora/checkpoint-300/tokenizer.json filter=lfs diff=lfs merge=lfs -text
93
+ v5/KTO/KTO_10k/MKTO_10k/tokenizer.json filter=lfs diff=lfs merge=lfs -text
94
+ v5/KTO/KTO_10k/lora/checkpoint-2300/tokenizer.json filter=lfs diff=lfs merge=lfs -text
95
+ v5/KTO/KTO_10k/lora/checkpoint-2400/tokenizer.json filter=lfs diff=lfs merge=lfs -text
96
+ v5/KTO/KTO_10k/lora/checkpoint-2500/tokenizer.json filter=lfs diff=lfs merge=lfs -text
97
+ v5/KTO/KTO_1k/MKTO_1k/tokenizer.json filter=lfs diff=lfs merge=lfs -text
98
+ v5/KTO/KTO_1k/lora/checkpoint-140/tokenizer.json filter=lfs diff=lfs merge=lfs -text
99
+ v5/KTO/KTO_1k/lora/checkpoint-240/tokenizer.json filter=lfs diff=lfs merge=lfs -text
100
+ v5/KTO/KTO_1k/lora/checkpoint-250/tokenizer.json filter=lfs diff=lfs merge=lfs -text
101
+ v5/KTO/KTO_20k/MKTO_20k/tokenizer.json filter=lfs diff=lfs merge=lfs -text
102
+ v5/KTO/KTO_20k/lora/checkpoint-4600/tokenizer.json filter=lfs diff=lfs merge=lfs -text
103
+ v5/KTO/KTO_20k/lora/checkpoint-4800/tokenizer.json filter=lfs diff=lfs merge=lfs -text
104
+ v5/KTO/KTO_20k/lora/checkpoint-5000/tokenizer.json filter=lfs diff=lfs merge=lfs -text
105
+ v5/KTO/KTO_2k/MKTO_2k/tokenizer.json filter=lfs diff=lfs merge=lfs -text
106
+ v5/KTO/KTO_2k/lora/checkpoint-180/tokenizer.json filter=lfs diff=lfs merge=lfs -text
107
+ v5/KTO/KTO_2k/lora/checkpoint-480/tokenizer.json filter=lfs diff=lfs merge=lfs -text
108
+ v5/KTO/KTO_2k/lora/checkpoint-500/tokenizer.json filter=lfs diff=lfs merge=lfs -text
109
+ v5/KTO/KTO_5k/MKTO_5k/tokenizer.json filter=lfs diff=lfs merge=lfs -text
110
+ v5/KTO/KTO_5k/lora/checkpoint-1150/tokenizer.json filter=lfs diff=lfs merge=lfs -text
111
+ v5/KTO/KTO_5k/lora/checkpoint-1200/tokenizer.json filter=lfs diff=lfs merge=lfs -text
112
+ v5/KTO/KTO_5k/lora/checkpoint-1250/tokenizer.json filter=lfs diff=lfs merge=lfs -text
113
+ v5/ORPO/ORPO_10k/MORPO_10k/tokenizer.json filter=lfs diff=lfs merge=lfs -text
114
+ v5/ORPO/ORPO_10k/lora/checkpoint-100/tokenizer.json filter=lfs diff=lfs merge=lfs -text
115
+ v5/ORPO/ORPO_10k/lora/checkpoint-2400/tokenizer.json filter=lfs diff=lfs merge=lfs -text
116
+ v5/ORPO/ORPO_10k/lora/checkpoint-2500/tokenizer.json filter=lfs diff=lfs merge=lfs -text
117
+ v5/ORPO/ORPO_1k/MORPO_1k/tokenizer.json filter=lfs diff=lfs merge=lfs -text
118
+ v5/ORPO/ORPO_1k/lora/checkpoint-240/tokenizer.json filter=lfs diff=lfs merge=lfs -text
119
+ v5/ORPO/ORPO_1k/lora/checkpoint-250/tokenizer.json filter=lfs diff=lfs merge=lfs -text
120
+ v5/ORPO/ORPO_1k/lora/checkpoint-30/tokenizer.json filter=lfs diff=lfs merge=lfs -text
121
+ v5/ORPO/ORPO_5k/MORPO_5k/tokenizer.json filter=lfs diff=lfs merge=lfs -text
122
+ v5/ORPO/ORPO_5k/lora/checkpoint-1200/tokenizer.json filter=lfs diff=lfs merge=lfs -text
123
+ v5/ORPO/ORPO_5k/lora/checkpoint-1250/tokenizer.json filter=lfs diff=lfs merge=lfs -text
124
+ v5/ORPO/ORPO_5k/lora/checkpoint-50/tokenizer.json filter=lfs diff=lfs merge=lfs -text
v5/DPO/DPO_10k/DPO_10k/README.md ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: meta-llama/Llama-3.2-1B-Instruct
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:meta-llama/Llama-3.2-1B-Instruct
7
+ - dpo
8
+ - lora
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
v5/DPO/DPO_10k/DPO_10k/adapter_config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.2-1B-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.1,
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": 64,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "o_proj",
33
+ "q_proj",
34
+ "down_proj",
35
+ "k_proj",
36
+ "gate_proj",
37
+ "up_proj",
38
+ "v_proj"
39
+ ],
40
+ "target_parameters": null,
41
+ "task_type": "CAUSAL_LM",
42
+ "trainable_token_indices": null,
43
+ "use_dora": false,
44
+ "use_qalora": false,
45
+ "use_rslora": false
46
+ }
v5/DPO/DPO_10k/DPO_10k/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1ecfca13898e10dceef60271a2e863d7cc7700ce1555e77092c7927008052ac
3
+ size 180385008
v5/DPO/DPO_10k/MDPO_10k/chat_template.jinja ADDED
@@ -0,0 +1,93 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ {%- if strftime_now is defined %}
10
+ {%- set date_string = strftime_now("%d %b %Y") %}
11
+ {%- else %}
12
+ {%- set date_string = "26 Jul 2024" %}
13
+ {%- endif %}
14
+ {%- endif %}
15
+ {%- if not tools is defined %}
16
+ {%- set tools = none %}
17
+ {%- endif %}
18
+
19
+ {#- This block extracts the system message, so we can slot it into the right place. #}
20
+ {%- if messages[0]['role'] == 'system' %}
21
+ {%- set system_message = messages[0]['content']|trim %}
22
+ {%- set messages = messages[1:] %}
23
+ {%- else %}
24
+ {%- set system_message = "" %}
25
+ {%- endif %}
26
+
27
+ {#- System message #}
28
+ {{- "<|start_header_id|>system<|end_header_id|>\n\n" }}
29
+ {%- if tools is not none %}
30
+ {{- "Environment: ipython\n" }}
31
+ {%- endif %}
32
+ {{- "Cutting Knowledge Date: December 2023\n" }}
33
+ {{- "Today Date: " + date_string + "\n\n" }}
34
+ {%- if tools is not none and not tools_in_user_message %}
35
+ {{- "You have access to the following functions. To call a function, please respond with JSON for a function call." }}
36
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
37
+ {{- "Do not use variables.\n\n" }}
38
+ {%- for t in tools %}
39
+ {{- t | tojson(indent=4) }}
40
+ {{- "\n\n" }}
41
+ {%- endfor %}
42
+ {%- endif %}
43
+ {{- system_message }}
44
+ {{- "<|eot_id|>" }}
45
+
46
+ {#- Custom tools are passed in a user message with some extra guidance #}
47
+ {%- if tools_in_user_message and not tools is none %}
48
+ {#- Extract the first user message so we can plug it in here #}
49
+ {%- if messages | length != 0 %}
50
+ {%- set first_user_message = messages[0]['content']|trim %}
51
+ {%- set messages = messages[1:] %}
52
+ {%- else %}
53
+ {{- raise_exception("Cannot put tools in the first user message when there's no first user message!") }}
54
+ {%- endif %}
55
+ {{- '<|start_header_id|>user<|end_header_id|>\n\n' -}}
56
+ {{- "Given the following functions, please respond with a JSON for a function call " }}
57
+ {{- "with its proper arguments that best answers the given prompt.\n\n" }}
58
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
59
+ {{- "Do not use variables.\n\n" }}
60
+ {%- for t in tools %}
61
+ {{- t | tojson(indent=4) }}
62
+ {{- "\n\n" }}
63
+ {%- endfor %}
64
+ {{- first_user_message + "<|eot_id|>"}}
65
+ {%- endif %}
66
+
67
+ {%- for message in messages %}
68
+ {%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}
69
+ {{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'+ message['content'] | trim + '<|eot_id|>' }}
70
+ {%- elif 'tool_calls' in message %}
71
+ {%- if not message.tool_calls|length == 1 %}
72
+ {{- raise_exception("This model only supports single tool-calls at once!") }}
73
+ {%- endif %}
74
+ {%- set tool_call = message.tool_calls[0].function %}
75
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' -}}
76
+ {{- '{"name": "' + tool_call.name + '", ' }}
77
+ {{- '"parameters": ' }}
78
+ {{- tool_call.arguments | tojson }}
79
+ {{- "}" }}
80
+ {{- "<|eot_id|>" }}
81
+ {%- elif message.role == "tool" or message.role == "ipython" %}
82
+ {{- "<|start_header_id|>ipython<|end_header_id|>\n\n" }}
83
+ {%- if message.content is mapping or message.content is iterable %}
84
+ {{- message.content | tojson }}
85
+ {%- else %}
86
+ {{- message.content }}
87
+ {%- endif %}
88
+ {{- "<|eot_id|>" }}
89
+ {%- endif %}
90
+ {%- endfor %}
91
+ {%- if add_generation_prompt %}
92
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' }}
93
+ {%- endif %}
v5/DPO/DPO_10k/MDPO_10k/config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "LlamaForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 128000,
8
+ "dtype": "float16",
9
+ "eos_token_id": [
10
+ 128001,
11
+ 128008,
12
+ 128009
13
+ ],
14
+ "head_dim": 64,
15
+ "hidden_act": "silu",
16
+ "hidden_size": 2048,
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 8192,
19
+ "max_position_embeddings": 131072,
20
+ "mlp_bias": false,
21
+ "model_type": "llama",
22
+ "num_attention_heads": 32,
23
+ "num_hidden_layers": 16,
24
+ "num_key_value_heads": 8,
25
+ "pad_token_id": null,
26
+ "pretraining_tp": 1,
27
+ "rms_norm_eps": 1e-05,
28
+ "rope_parameters": {
29
+ "factor": 32.0,
30
+ "high_freq_factor": 4.0,
31
+ "low_freq_factor": 1.0,
32
+ "original_max_position_embeddings": 8192,
33
+ "rope_theta": 500000.0,
34
+ "rope_type": "llama3"
35
+ },
36
+ "tie_word_embeddings": true,
37
+ "transformers_version": "5.0.0",
38
+ "use_cache": true,
39
+ "vocab_size": 128256
40
+ }
v5/DPO/DPO_10k/MDPO_10k/generation_config.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 128000,
3
+ "do_sample": true,
4
+ "eos_token_id": [
5
+ 128001,
6
+ 128008,
7
+ 128009
8
+ ],
9
+ "temperature": 0.6,
10
+ "top_p": 0.9,
11
+ "transformers_version": "5.0.0"
12
+ }
v5/DPO/DPO_10k/MDPO_10k/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c840c13f70893c0b8d55e3acd6553093375b17eaa47b583753de8b121643123
3
+ size 2471645464
v5/DPO/DPO_10k/MDPO_10k/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b9e4e7fb171f92fd137b777cc2714bf87d11576700a1dcd7a399e7bbe39537b
3
+ size 17209920
v5/DPO/DPO_10k/MDPO_10k/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
+ }
v5/DPO/DPO_10k/lora/README.md ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: meta-llama/Llama-3.2-1B-Instruct
3
+ library_name: transformers
4
+ model_name: lora
5
+ tags:
6
+ - generated_from_trainer
7
+ - dpo
8
+ - trl
9
+ licence: license
10
+ ---
11
+
12
+ # Model Card for lora
13
+
14
+ This model is a fine-tuned version of [meta-llama/Llama-3.2-1B-Instruct](https://huggingface.co/meta-llama/Llama-3.2-1B-Instruct).
15
+ It has been trained using [TRL](https://github.com/huggingface/trl).
16
+
17
+ ## Quick start
18
+
19
+ ```python
20
+ from transformers import pipeline
21
+
22
+ 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?"
23
+ generator = pipeline("text-generation", model="None", device="cuda")
24
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
25
+ print(output["generated_text"])
26
+ ```
27
+
28
+ ## Training procedure
29
+
30
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/sea-rod/huggingface/runs/5573bwv9)
31
+
32
+
33
+ This model was trained with DPO, a method introduced in [Direct Preference Optimization: Your Language Model is Secretly a Reward Model](https://huggingface.co/papers/2305.18290).
34
+
35
+ ### Framework versions
36
+
37
+ - TRL: 0.27.2
38
+ - Transformers: 5.0.0
39
+ - Pytorch: 2.8.0+cu128
40
+ - Datasets: 4.5.0
41
+ - Tokenizers: 0.22.2
42
+
43
+ ## Citations
44
+
45
+ Cite DPO as:
46
+
47
+ ```bibtex
48
+ @inproceedings{rafailov2023direct,
49
+ title = {{Direct Preference Optimization: Your Language Model is Secretly a Reward Model}},
50
+ author = {Rafael Rafailov and Archit Sharma and Eric Mitchell and Christopher D. Manning and Stefano Ermon and Chelsea Finn},
51
+ year = 2023,
52
+ booktitle = {Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, NeurIPS 2023, New Orleans, LA, USA, December 10 - 16, 2023},
53
+ url = {http://papers.nips.cc/paper_files/paper/2023/hash/a85b405ed65c6477a4fe8302b5e06ce7-Abstract-Conference.html},
54
+ editor = {Alice Oh and Tristan Naumann and Amir Globerson and Kate Saenko and Moritz Hardt and Sergey Levine},
55
+ }
56
+ ```
57
+
58
+ Cite TRL as:
59
+
60
+ ```bibtex
61
+ @misc{vonwerra2022trl,
62
+ title = {{TRL: Transformer Reinforcement Learning}},
63
+ author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallou{\'e}dec},
64
+ year = 2020,
65
+ journal = {GitHub repository},
66
+ publisher = {GitHub},
67
+ howpublished = {\url{https://github.com/huggingface/trl}}
68
+ }
69
+ ```
v5/DPO/DPO_10k/lora/checkpoint-1300/README.md ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: meta-llama/Llama-3.2-1B-Instruct
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:meta-llama/Llama-3.2-1B-Instruct
7
+ - dpo
8
+ - lora
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
v5/DPO/DPO_10k/lora/checkpoint-1300/adapter_config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.2-1B-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.1,
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": 64,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "o_proj",
33
+ "q_proj",
34
+ "down_proj",
35
+ "k_proj",
36
+ "gate_proj",
37
+ "up_proj",
38
+ "v_proj"
39
+ ],
40
+ "target_parameters": null,
41
+ "task_type": "CAUSAL_LM",
42
+ "trainable_token_indices": null,
43
+ "use_dora": false,
44
+ "use_qalora": false,
45
+ "use_rslora": false
46
+ }
v5/DPO/DPO_10k/lora/checkpoint-1300/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1ecfca13898e10dceef60271a2e863d7cc7700ce1555e77092c7927008052ac
3
+ size 180385008
v5/DPO/DPO_10k/lora/checkpoint-1300/chat_template.jinja ADDED
@@ -0,0 +1,93 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ {%- if strftime_now is defined %}
10
+ {%- set date_string = strftime_now("%d %b %Y") %}
11
+ {%- else %}
12
+ {%- set date_string = "26 Jul 2024" %}
13
+ {%- endif %}
14
+ {%- endif %}
15
+ {%- if not tools is defined %}
16
+ {%- set tools = none %}
17
+ {%- endif %}
18
+
19
+ {#- This block extracts the system message, so we can slot it into the right place. #}
20
+ {%- if messages[0]['role'] == 'system' %}
21
+ {%- set system_message = messages[0]['content']|trim %}
22
+ {%- set messages = messages[1:] %}
23
+ {%- else %}
24
+ {%- set system_message = "" %}
25
+ {%- endif %}
26
+
27
+ {#- System message #}
28
+ {{- "<|start_header_id|>system<|end_header_id|>\n\n" }}
29
+ {%- if tools is not none %}
30
+ {{- "Environment: ipython\n" }}
31
+ {%- endif %}
32
+ {{- "Cutting Knowledge Date: December 2023\n" }}
33
+ {{- "Today Date: " + date_string + "\n\n" }}
34
+ {%- if tools is not none and not tools_in_user_message %}
35
+ {{- "You have access to the following functions. To call a function, please respond with JSON for a function call." }}
36
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
37
+ {{- "Do not use variables.\n\n" }}
38
+ {%- for t in tools %}
39
+ {{- t | tojson(indent=4) }}
40
+ {{- "\n\n" }}
41
+ {%- endfor %}
42
+ {%- endif %}
43
+ {{- system_message }}
44
+ {{- "<|eot_id|>" }}
45
+
46
+ {#- Custom tools are passed in a user message with some extra guidance #}
47
+ {%- if tools_in_user_message and not tools is none %}
48
+ {#- Extract the first user message so we can plug it in here #}
49
+ {%- if messages | length != 0 %}
50
+ {%- set first_user_message = messages[0]['content']|trim %}
51
+ {%- set messages = messages[1:] %}
52
+ {%- else %}
53
+ {{- raise_exception("Cannot put tools in the first user message when there's no first user message!") }}
54
+ {%- endif %}
55
+ {{- '<|start_header_id|>user<|end_header_id|>\n\n' -}}
56
+ {{- "Given the following functions, please respond with a JSON for a function call " }}
57
+ {{- "with its proper arguments that best answers the given prompt.\n\n" }}
58
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
59
+ {{- "Do not use variables.\n\n" }}
60
+ {%- for t in tools %}
61
+ {{- t | tojson(indent=4) }}
62
+ {{- "\n\n" }}
63
+ {%- endfor %}
64
+ {{- first_user_message + "<|eot_id|>"}}
65
+ {%- endif %}
66
+
67
+ {%- for message in messages %}
68
+ {%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}
69
+ {{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'+ message['content'] | trim + '<|eot_id|>' }}
70
+ {%- elif 'tool_calls' in message %}
71
+ {%- if not message.tool_calls|length == 1 %}
72
+ {{- raise_exception("This model only supports single tool-calls at once!") }}
73
+ {%- endif %}
74
+ {%- set tool_call = message.tool_calls[0].function %}
75
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' -}}
76
+ {{- '{"name": "' + tool_call.name + '", ' }}
77
+ {{- '"parameters": ' }}
78
+ {{- tool_call.arguments | tojson }}
79
+ {{- "}" }}
80
+ {{- "<|eot_id|>" }}
81
+ {%- elif message.role == "tool" or message.role == "ipython" %}
82
+ {{- "<|start_header_id|>ipython<|end_header_id|>\n\n" }}
83
+ {%- if message.content is mapping or message.content is iterable %}
84
+ {{- message.content | tojson }}
85
+ {%- else %}
86
+ {{- message.content }}
87
+ {%- endif %}
88
+ {{- "<|eot_id|>" }}
89
+ {%- endif %}
90
+ {%- endfor %}
91
+ {%- if add_generation_prompt %}
92
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' }}
93
+ {%- endif %}
v5/DPO/DPO_10k/lora/checkpoint-1300/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71b5e638f59b9595d8dca87dcaa601984e35f6d6732e3de54770d1173653d04e
3
+ size 360902475
v5/DPO/DPO_10k/lora/checkpoint-1300/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:95d744506ed8242dbe82c0f3357716f73248e5153ff68604326958faa28d9296
3
+ size 14645
v5/DPO/DPO_10k/lora/checkpoint-1300/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5e6dc9658957f5cc4ceed4243b2de1cacb7762930be64bb5a8e59e057d65e5e2
3
+ size 1383
v5/DPO/DPO_10k/lora/checkpoint-1300/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:619902af4750c8c03b81f910de02f255dd9ba762e9ef95909c545df05bfb4a75
3
+ size 1465
v5/DPO/DPO_10k/lora/checkpoint-1300/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b9e4e7fb171f92fd137b777cc2714bf87d11576700a1dcd7a399e7bbe39537b
3
+ size 17209920
v5/DPO/DPO_10k/lora/checkpoint-1300/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
+ }
v5/DPO/DPO_10k/lora/checkpoint-1300/trainer_state.json ADDED
@@ -0,0 +1,2192 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 1300,
3
+ "best_metric": 0.5460000038146973,
4
+ "best_model_checkpoint": "output/lora/checkpoint-1300",
5
+ "epoch": 1.04,
6
+ "eval_steps": 100,
7
+ "global_step": 1300,
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.008,
14
+ "grad_norm": 4.452983379364014,
15
+ "learning_rate": 3.6e-08,
16
+ "logits/chosen": 1.5510008335113525,
17
+ "logits/rejected": 1.5244438648223877,
18
+ "logps/chosen": -131.24708557128906,
19
+ "logps/rejected": -146.8297576904297,
20
+ "loss": 0.6932957172393799,
21
+ "rewards/accuracies": 0.1875,
22
+ "rewards/chosen": -0.00019072293071076274,
23
+ "rewards/margins": -0.00029331922996789217,
24
+ "rewards/rejected": 0.0001025962847052142,
25
+ "step": 10
26
+ },
27
+ {
28
+ "epoch": 0.016,
29
+ "grad_norm": 5.203515529632568,
30
+ "learning_rate": 7.599999999999999e-08,
31
+ "logits/chosen": 1.6611576080322266,
32
+ "logits/rejected": 1.6220839023590088,
33
+ "logps/chosen": -156.2080078125,
34
+ "logps/rejected": -142.72964477539062,
35
+ "loss": 0.6937986850738526,
36
+ "rewards/accuracies": 0.4625000059604645,
37
+ "rewards/chosen": -0.0006145072402432561,
38
+ "rewards/margins": -0.0012890815269201994,
39
+ "rewards/rejected": 0.000674574519507587,
40
+ "step": 20
41
+ },
42
+ {
43
+ "epoch": 0.024,
44
+ "grad_norm": 4.087289810180664,
45
+ "learning_rate": 1.16e-07,
46
+ "logits/chosen": 1.9773778915405273,
47
+ "logits/rejected": 1.8304665088653564,
48
+ "logps/chosen": -163.54708862304688,
49
+ "logps/rejected": -157.88926696777344,
50
+ "loss": 0.6931437492370606,
51
+ "rewards/accuracies": 0.5249999761581421,
52
+ "rewards/chosen": 0.0003584886435419321,
53
+ "rewards/margins": 2.1700874640373513e-05,
54
+ "rewards/rejected": 0.00033678775071166456,
55
+ "step": 30
56
+ },
57
+ {
58
+ "epoch": 0.032,
59
+ "grad_norm": 4.74172830581665,
60
+ "learning_rate": 1.56e-07,
61
+ "logits/chosen": 1.5896285772323608,
62
+ "logits/rejected": 1.7109922170639038,
63
+ "logps/chosen": -144.44276428222656,
64
+ "logps/rejected": -133.09629821777344,
65
+ "loss": 0.6932558059692383,
66
+ "rewards/accuracies": 0.4749999940395355,
67
+ "rewards/chosen": 0.00048673158744350076,
68
+ "rewards/margins": -0.00020531899644993246,
69
+ "rewards/rejected": 0.0006920504383742809,
70
+ "step": 40
71
+ },
72
+ {
73
+ "epoch": 0.04,
74
+ "grad_norm": 4.32133150100708,
75
+ "learning_rate": 1.96e-07,
76
+ "logits/chosen": 1.5152148008346558,
77
+ "logits/rejected": 1.585367202758789,
78
+ "logps/chosen": -131.73226928710938,
79
+ "logps/rejected": -136.8301239013672,
80
+ "loss": 0.6930522918701172,
81
+ "rewards/accuracies": 0.48750001192092896,
82
+ "rewards/chosen": 0.0008355070021934807,
83
+ "rewards/margins": 0.00019948731642216444,
84
+ "rewards/rejected": 0.0006360196857713163,
85
+ "step": 50
86
+ },
87
+ {
88
+ "epoch": 0.048,
89
+ "grad_norm": 3.915316343307495,
90
+ "learning_rate": 2.3599999999999997e-07,
91
+ "logits/chosen": 1.5138778686523438,
92
+ "logits/rejected": 1.4824903011322021,
93
+ "logps/chosen": -136.43399047851562,
94
+ "logps/rejected": -126.70623779296875,
95
+ "loss": 0.6929163455963134,
96
+ "rewards/accuracies": 0.550000011920929,
97
+ "rewards/chosen": 0.000302538916002959,
98
+ "rewards/margins": 0.00047383070341311395,
99
+ "rewards/rejected": -0.00017129186016973108,
100
+ "step": 60
101
+ },
102
+ {
103
+ "epoch": 0.056,
104
+ "grad_norm": 4.329769134521484,
105
+ "learning_rate": 2.7600000000000004e-07,
106
+ "logits/chosen": 1.6920913457870483,
107
+ "logits/rejected": 1.8169019222259521,
108
+ "logps/chosen": -152.056640625,
109
+ "logps/rejected": -155.9404296875,
110
+ "loss": 0.6935575008392334,
111
+ "rewards/accuracies": 0.512499988079071,
112
+ "rewards/chosen": 0.0025545789394527674,
113
+ "rewards/margins": -0.0008073424687609076,
114
+ "rewards/rejected": 0.003361921291798353,
115
+ "step": 70
116
+ },
117
+ {
118
+ "epoch": 0.064,
119
+ "grad_norm": 3.964193344116211,
120
+ "learning_rate": 3.1599999999999997e-07,
121
+ "logits/chosen": 1.5645431280136108,
122
+ "logits/rejected": 1.5879082679748535,
123
+ "logps/chosen": -147.78839111328125,
124
+ "logps/rejected": -135.19906616210938,
125
+ "loss": 0.6925086498260498,
126
+ "rewards/accuracies": 0.6000000238418579,
127
+ "rewards/chosen": 0.006093275733292103,
128
+ "rewards/margins": 0.0012893510283902287,
129
+ "rewards/rejected": 0.004803924821317196,
130
+ "step": 80
131
+ },
132
+ {
133
+ "epoch": 0.072,
134
+ "grad_norm": 3.935694694519043,
135
+ "learning_rate": 3.5599999999999996e-07,
136
+ "logits/chosen": 1.5960246324539185,
137
+ "logits/rejected": 1.6901094913482666,
138
+ "logps/chosen": -157.85256958007812,
139
+ "logps/rejected": -150.51974487304688,
140
+ "loss": 0.6931850433349609,
141
+ "rewards/accuracies": 0.42500001192092896,
142
+ "rewards/chosen": 0.010219026356935501,
143
+ "rewards/margins": -5.4271204135147855e-05,
144
+ "rewards/rejected": 0.010273297317326069,
145
+ "step": 90
146
+ },
147
+ {
148
+ "epoch": 0.08,
149
+ "grad_norm": 5.13019323348999,
150
+ "learning_rate": 3.96e-07,
151
+ "logits/chosen": 1.5446935892105103,
152
+ "logits/rejected": 1.6452451944351196,
153
+ "logps/chosen": -149.88038635253906,
154
+ "logps/rejected": -169.9078826904297,
155
+ "loss": 0.6935123443603516,
156
+ "rewards/accuracies": 0.5,
157
+ "rewards/chosen": 0.011777262203395367,
158
+ "rewards/margins": -0.0007071519503369927,
159
+ "rewards/rejected": 0.0124844154343009,
160
+ "step": 100
161
+ },
162
+ {
163
+ "epoch": 0.08,
164
+ "eval_logits/chosen": 1.5421667098999023,
165
+ "eval_logits/rejected": 1.5734084844589233,
166
+ "eval_logps/chosen": -153.21649169921875,
167
+ "eval_logps/rejected": -147.7952117919922,
168
+ "eval_loss": 0.6929068565368652,
169
+ "eval_rewards/accuracies": 0.5180000066757202,
170
+ "eval_rewards/chosen": 0.011265883222222328,
171
+ "eval_rewards/margins": 0.0005034058121964335,
172
+ "eval_rewards/rejected": 0.010762478224933147,
173
+ "eval_runtime": 90.2131,
174
+ "eval_samples_per_second": 5.542,
175
+ "eval_steps_per_second": 2.771,
176
+ "step": 100
177
+ },
178
+ {
179
+ "epoch": 0.088,
180
+ "grad_norm": 5.695896625518799,
181
+ "learning_rate": 4.36e-07,
182
+ "logits/chosen": 1.781393051147461,
183
+ "logits/rejected": 1.7461833953857422,
184
+ "logps/chosen": -172.24188232421875,
185
+ "logps/rejected": -154.40878295898438,
186
+ "loss": 0.6922736167907715,
187
+ "rewards/accuracies": 0.574999988079071,
188
+ "rewards/chosen": 0.014022141695022583,
189
+ "rewards/margins": 0.001777560799382627,
190
+ "rewards/rejected": 0.012244580313563347,
191
+ "step": 110
192
+ },
193
+ {
194
+ "epoch": 0.096,
195
+ "grad_norm": 4.398581027984619,
196
+ "learning_rate": 4.76e-07,
197
+ "logits/chosen": 1.611268401145935,
198
+ "logits/rejected": 1.6106624603271484,
199
+ "logps/chosen": -135.13426208496094,
200
+ "logps/rejected": -139.7284393310547,
201
+ "loss": 0.6927696228027344,
202
+ "rewards/accuracies": 0.5625,
203
+ "rewards/chosen": 0.014869053848087788,
204
+ "rewards/margins": 0.0007819391903467476,
205
+ "rewards/rejected": 0.014087115414440632,
206
+ "step": 120
207
+ },
208
+ {
209
+ "epoch": 0.104,
210
+ "grad_norm": 5.19202995300293,
211
+ "learning_rate": 5.16e-07,
212
+ "logits/chosen": 1.5615273714065552,
213
+ "logits/rejected": 1.7724416255950928,
214
+ "logps/chosen": -157.66746520996094,
215
+ "logps/rejected": -161.90391540527344,
216
+ "loss": 0.6928309917449951,
217
+ "rewards/accuracies": 0.5625,
218
+ "rewards/chosen": 0.013302234932780266,
219
+ "rewards/margins": 0.0006531739491038024,
220
+ "rewards/rejected": 0.012649061158299446,
221
+ "step": 130
222
+ },
223
+ {
224
+ "epoch": 0.112,
225
+ "grad_norm": 3.4575726985931396,
226
+ "learning_rate": 5.560000000000001e-07,
227
+ "logits/chosen": 1.492018699645996,
228
+ "logits/rejected": 1.5187314748764038,
229
+ "logps/chosen": -131.4152374267578,
230
+ "logps/rejected": -125.62705993652344,
231
+ "loss": 0.6929276943206787,
232
+ "rewards/accuracies": 0.5249999761581421,
233
+ "rewards/chosen": 0.014355423860251904,
234
+ "rewards/margins": 0.000462935131508857,
235
+ "rewards/rejected": 0.013892488554120064,
236
+ "step": 140
237
+ },
238
+ {
239
+ "epoch": 0.12,
240
+ "grad_norm": 4.500187397003174,
241
+ "learning_rate": 5.96e-07,
242
+ "logits/chosen": 1.5862048864364624,
243
+ "logits/rejected": 1.6784181594848633,
244
+ "logps/chosen": -163.6667938232422,
245
+ "logps/rejected": -157.76402282714844,
246
+ "loss": 0.6910766124725342,
247
+ "rewards/accuracies": 0.6875,
248
+ "rewards/chosen": 0.018648523837327957,
249
+ "rewards/margins": 0.004183619283139706,
250
+ "rewards/rejected": 0.014464902691543102,
251
+ "step": 150
252
+ },
253
+ {
254
+ "epoch": 0.128,
255
+ "grad_norm": 4.113079071044922,
256
+ "learning_rate": 6.36e-07,
257
+ "logits/chosen": 1.7717370986938477,
258
+ "logits/rejected": 1.8070589303970337,
259
+ "logps/chosen": -158.02734375,
260
+ "logps/rejected": -145.92495727539062,
261
+ "loss": 0.6927172183990479,
262
+ "rewards/accuracies": 0.48750001192092896,
263
+ "rewards/chosen": 0.02493301033973694,
264
+ "rewards/margins": 0.0009325124556198716,
265
+ "rewards/rejected": 0.024000495672225952,
266
+ "step": 160
267
+ },
268
+ {
269
+ "epoch": 0.136,
270
+ "grad_norm": 5.4983696937561035,
271
+ "learning_rate": 6.76e-07,
272
+ "logits/chosen": 1.5733931064605713,
273
+ "logits/rejected": 1.6008774042129517,
274
+ "logps/chosen": -147.1856689453125,
275
+ "logps/rejected": -159.93077087402344,
276
+ "loss": 0.6926907062530517,
277
+ "rewards/accuracies": 0.512499988079071,
278
+ "rewards/chosen": 0.02877199277281761,
279
+ "rewards/margins": 0.0010106085101142526,
280
+ "rewards/rejected": 0.027761384844779968,
281
+ "step": 170
282
+ },
283
+ {
284
+ "epoch": 0.144,
285
+ "grad_norm": 4.50191068649292,
286
+ "learning_rate": 7.159999999999999e-07,
287
+ "logits/chosen": 1.5886398553848267,
288
+ "logits/rejected": 1.7399513721466064,
289
+ "logps/chosen": -157.6659393310547,
290
+ "logps/rejected": -160.65431213378906,
291
+ "loss": 0.6925735473632812,
292
+ "rewards/accuracies": 0.5,
293
+ "rewards/chosen": 0.024526067078113556,
294
+ "rewards/margins": 0.0012190367560833693,
295
+ "rewards/rejected": 0.02330702915787697,
296
+ "step": 180
297
+ },
298
+ {
299
+ "epoch": 0.152,
300
+ "grad_norm": 4.708652019500732,
301
+ "learning_rate": 7.559999999999999e-07,
302
+ "logits/chosen": 1.6504443883895874,
303
+ "logits/rejected": 1.7761001586914062,
304
+ "logps/chosen": -141.34536743164062,
305
+ "logps/rejected": -143.41159057617188,
306
+ "loss": 0.6928653240203857,
307
+ "rewards/accuracies": 0.5249999761581421,
308
+ "rewards/chosen": 0.026504456996917725,
309
+ "rewards/margins": 0.0006459927535615861,
310
+ "rewards/rejected": 0.02585846558213234,
311
+ "step": 190
312
+ },
313
+ {
314
+ "epoch": 0.16,
315
+ "grad_norm": 4.143187046051025,
316
+ "learning_rate": 7.96e-07,
317
+ "logits/chosen": 1.7108211517333984,
318
+ "logits/rejected": 1.6271438598632812,
319
+ "logps/chosen": -158.04931640625,
320
+ "logps/rejected": -132.23463439941406,
321
+ "loss": 0.6927096843719482,
322
+ "rewards/accuracies": 0.4625000059604645,
323
+ "rewards/chosen": 0.02748889848589897,
324
+ "rewards/margins": 0.0009879134595394135,
325
+ "rewards/rejected": 0.026500985026359558,
326
+ "step": 200
327
+ },
328
+ {
329
+ "epoch": 0.16,
330
+ "eval_logits/chosen": 1.54364013671875,
331
+ "eval_logits/rejected": 1.5745173692703247,
332
+ "eval_logps/chosen": -153.04653930664062,
333
+ "eval_logps/rejected": -147.63844299316406,
334
+ "eval_loss": 0.6923084855079651,
335
+ "eval_rewards/accuracies": 0.5139999985694885,
336
+ "eval_rewards/chosen": 0.028259562328457832,
337
+ "eval_rewards/margins": 0.0018185621593147516,
338
+ "eval_rewards/rejected": 0.026441000401973724,
339
+ "eval_runtime": 90.4481,
340
+ "eval_samples_per_second": 5.528,
341
+ "eval_steps_per_second": 2.764,
342
+ "step": 200
343
+ },
344
+ {
345
+ "epoch": 0.168,
346
+ "grad_norm": 4.559652328491211,
347
+ "learning_rate": 8.359999999999999e-07,
348
+ "logits/chosen": 1.7295278310775757,
349
+ "logits/rejected": 1.6801897287368774,
350
+ "logps/chosen": -158.0893096923828,
351
+ "logps/rejected": -168.72427368164062,
352
+ "loss": 0.6922987461090088,
353
+ "rewards/accuracies": 0.5249999761581421,
354
+ "rewards/chosen": 0.02837887406349182,
355
+ "rewards/margins": 0.0018027331680059433,
356
+ "rewards/rejected": 0.026576142758131027,
357
+ "step": 210
358
+ },
359
+ {
360
+ "epoch": 0.176,
361
+ "grad_norm": 3.907545328140259,
362
+ "learning_rate": 8.76e-07,
363
+ "logits/chosen": 1.6849712133407593,
364
+ "logits/rejected": 1.7441444396972656,
365
+ "logps/chosen": -158.67384338378906,
366
+ "logps/rejected": -143.02920532226562,
367
+ "loss": 0.6933117389678956,
368
+ "rewards/accuracies": 0.550000011920929,
369
+ "rewards/chosen": 0.03286944702267647,
370
+ "rewards/margins": -0.0002176974667236209,
371
+ "rewards/rejected": 0.03308714181184769,
372
+ "step": 220
373
+ },
374
+ {
375
+ "epoch": 0.184,
376
+ "grad_norm": 3.5083253383636475,
377
+ "learning_rate": 9.16e-07,
378
+ "logits/chosen": 1.5701725482940674,
379
+ "logits/rejected": 1.7182495594024658,
380
+ "logps/chosen": -160.56790161132812,
381
+ "logps/rejected": -138.05374145507812,
382
+ "loss": 0.6915814399719238,
383
+ "rewards/accuracies": 0.5375000238418579,
384
+ "rewards/chosen": 0.03840702772140503,
385
+ "rewards/margins": 0.0033035180531442165,
386
+ "rewards/rejected": 0.035103507339954376,
387
+ "step": 230
388
+ },
389
+ {
390
+ "epoch": 0.192,
391
+ "grad_norm": 4.424270153045654,
392
+ "learning_rate": 9.559999999999998e-07,
393
+ "logits/chosen": 1.671190857887268,
394
+ "logits/rejected": 1.6964585781097412,
395
+ "logps/chosen": -170.28260803222656,
396
+ "logps/rejected": -144.33534240722656,
397
+ "loss": 0.6900368690490722,
398
+ "rewards/accuracies": 0.5249999761581421,
399
+ "rewards/chosen": 0.0511443130671978,
400
+ "rewards/margins": 0.006511001847684383,
401
+ "rewards/rejected": 0.04463331401348114,
402
+ "step": 240
403
+ },
404
+ {
405
+ "epoch": 0.2,
406
+ "grad_norm": 4.5393967628479,
407
+ "learning_rate": 9.959999999999999e-07,
408
+ "logits/chosen": 1.6391950845718384,
409
+ "logits/rejected": 1.5815935134887695,
410
+ "logps/chosen": -160.45225524902344,
411
+ "logps/rejected": -147.56185913085938,
412
+ "loss": 0.6940414905548096,
413
+ "rewards/accuracies": 0.44999998807907104,
414
+ "rewards/chosen": 0.04959743469953537,
415
+ "rewards/margins": -0.001381749869324267,
416
+ "rewards/rejected": 0.05097918584942818,
417
+ "step": 250
418
+ },
419
+ {
420
+ "epoch": 0.208,
421
+ "grad_norm": 4.256033897399902,
422
+ "learning_rate": 9.959999999999999e-07,
423
+ "logits/chosen": 1.5204452276229858,
424
+ "logits/rejected": 1.6171140670776367,
425
+ "logps/chosen": -131.5397186279297,
426
+ "logps/rejected": -145.2186279296875,
427
+ "loss": 0.6930822372436524,
428
+ "rewards/accuracies": 0.4625000059604645,
429
+ "rewards/chosen": 0.045291412621736526,
430
+ "rewards/margins": 0.000405142258387059,
431
+ "rewards/rejected": 0.0448862686753273,
432
+ "step": 260
433
+ },
434
+ {
435
+ "epoch": 0.216,
436
+ "grad_norm": 4.027031421661377,
437
+ "learning_rate": 9.915555555555556e-07,
438
+ "logits/chosen": 1.6407123804092407,
439
+ "logits/rejected": 1.7262824773788452,
440
+ "logps/chosen": -145.78701782226562,
441
+ "logps/rejected": -146.34481811523438,
442
+ "loss": 0.6946187496185303,
443
+ "rewards/accuracies": 0.512499988079071,
444
+ "rewards/chosen": 0.04104622080922127,
445
+ "rewards/margins": -0.0027089794166386127,
446
+ "rewards/rejected": 0.043755196034908295,
447
+ "step": 270
448
+ },
449
+ {
450
+ "epoch": 0.224,
451
+ "grad_norm": 5.568243026733398,
452
+ "learning_rate": 9.87111111111111e-07,
453
+ "logits/chosen": 1.6697533130645752,
454
+ "logits/rejected": 1.5154677629470825,
455
+ "logps/chosen": -171.0277099609375,
456
+ "logps/rejected": -154.05654907226562,
457
+ "loss": 0.6873753070831299,
458
+ "rewards/accuracies": 0.6499999761581421,
459
+ "rewards/chosen": 0.04779375344514847,
460
+ "rewards/margins": 0.011815806850790977,
461
+ "rewards/rejected": 0.03597795218229294,
462
+ "step": 280
463
+ },
464
+ {
465
+ "epoch": 0.232,
466
+ "grad_norm": 4.041477680206299,
467
+ "learning_rate": 9.826666666666667e-07,
468
+ "logits/chosen": 1.6633354425430298,
469
+ "logits/rejected": 1.6905081272125244,
470
+ "logps/chosen": -142.88864135742188,
471
+ "logps/rejected": -152.04757690429688,
472
+ "loss": 0.6929869174957275,
473
+ "rewards/accuracies": 0.512499988079071,
474
+ "rewards/chosen": 0.04824609309434891,
475
+ "rewards/margins": 0.0005596639821305871,
476
+ "rewards/rejected": 0.04768642783164978,
477
+ "step": 290
478
+ },
479
+ {
480
+ "epoch": 0.24,
481
+ "grad_norm": 4.9481635093688965,
482
+ "learning_rate": 9.782222222222222e-07,
483
+ "logits/chosen": 1.617485761642456,
484
+ "logits/rejected": 1.6837307214736938,
485
+ "logps/chosen": -152.088134765625,
486
+ "logps/rejected": -164.15158081054688,
487
+ "loss": 0.6897455215454101,
488
+ "rewards/accuracies": 0.612500011920929,
489
+ "rewards/chosen": 0.057097338140010834,
490
+ "rewards/margins": 0.007080497685819864,
491
+ "rewards/rejected": 0.050016842782497406,
492
+ "step": 300
493
+ },
494
+ {
495
+ "epoch": 0.24,
496
+ "eval_logits/chosen": 1.5520602464675903,
497
+ "eval_logits/rejected": 1.5826917886734009,
498
+ "eval_logps/chosen": -152.8227996826172,
499
+ "eval_logps/rejected": -147.43496704101562,
500
+ "eval_loss": 0.6914217472076416,
501
+ "eval_rewards/accuracies": 0.5260000228881836,
502
+ "eval_rewards/chosen": 0.05063560605049133,
503
+ "eval_rewards/margins": 0.0038486982230097055,
504
+ "eval_rewards/rejected": 0.04678690433502197,
505
+ "eval_runtime": 90.237,
506
+ "eval_samples_per_second": 5.541,
507
+ "eval_steps_per_second": 2.77,
508
+ "step": 300
509
+ },
510
+ {
511
+ "epoch": 0.248,
512
+ "grad_norm": 4.486109256744385,
513
+ "learning_rate": 9.737777777777777e-07,
514
+ "logits/chosen": 1.7188745737075806,
515
+ "logits/rejected": 1.7590553760528564,
516
+ "logps/chosen": -140.6877899169922,
517
+ "logps/rejected": -155.31893920898438,
518
+ "loss": 0.6952545166015625,
519
+ "rewards/accuracies": 0.38749998807907104,
520
+ "rewards/chosen": 0.046565137803554535,
521
+ "rewards/margins": -0.003889651270583272,
522
+ "rewards/rejected": 0.050454795360565186,
523
+ "step": 310
524
+ },
525
+ {
526
+ "epoch": 0.256,
527
+ "grad_norm": 4.374355316162109,
528
+ "learning_rate": 9.693333333333334e-07,
529
+ "logits/chosen": 1.7103513479232788,
530
+ "logits/rejected": 1.7379261255264282,
531
+ "logps/chosen": -137.8660888671875,
532
+ "logps/rejected": -140.40956115722656,
533
+ "loss": 0.692354393005371,
534
+ "rewards/accuracies": 0.5249999761581421,
535
+ "rewards/chosen": 0.04029277712106705,
536
+ "rewards/margins": 0.0018124934285879135,
537
+ "rewards/rejected": 0.038480278104543686,
538
+ "step": 320
539
+ },
540
+ {
541
+ "epoch": 0.264,
542
+ "grad_norm": 4.618821144104004,
543
+ "learning_rate": 9.648888888888889e-07,
544
+ "logits/chosen": 1.5603777170181274,
545
+ "logits/rejected": 1.5868213176727295,
546
+ "logps/chosen": -157.3379669189453,
547
+ "logps/rejected": -182.77377319335938,
548
+ "loss": 0.6921377182006836,
549
+ "rewards/accuracies": 0.5249999761581421,
550
+ "rewards/chosen": 0.03832743316888809,
551
+ "rewards/margins": 0.0022720033302903175,
552
+ "rewards/rejected": 0.03605542704463005,
553
+ "step": 330
554
+ },
555
+ {
556
+ "epoch": 0.272,
557
+ "grad_norm": 4.1974406242370605,
558
+ "learning_rate": 9.604444444444443e-07,
559
+ "logits/chosen": 1.8520517349243164,
560
+ "logits/rejected": 1.716774582862854,
561
+ "logps/chosen": -158.16363525390625,
562
+ "logps/rejected": -149.66162109375,
563
+ "loss": 0.692081069946289,
564
+ "rewards/accuracies": 0.5874999761581421,
565
+ "rewards/chosen": 0.0396885983645916,
566
+ "rewards/margins": 0.0023556998930871487,
567
+ "rewards/rejected": 0.03733289986848831,
568
+ "step": 340
569
+ },
570
+ {
571
+ "epoch": 0.28,
572
+ "grad_norm": 4.700806140899658,
573
+ "learning_rate": 9.559999999999998e-07,
574
+ "logits/chosen": 1.6886537075042725,
575
+ "logits/rejected": 1.8079423904418945,
576
+ "logps/chosen": -165.5438232421875,
577
+ "logps/rejected": -194.98428344726562,
578
+ "loss": 0.6890993118286133,
579
+ "rewards/accuracies": 0.612500011920929,
580
+ "rewards/chosen": 0.05031610652804375,
581
+ "rewards/margins": 0.008422226645052433,
582
+ "rewards/rejected": 0.041893888264894485,
583
+ "step": 350
584
+ },
585
+ {
586
+ "epoch": 0.288,
587
+ "grad_norm": 3.7786731719970703,
588
+ "learning_rate": 9.515555555555555e-07,
589
+ "logits/chosen": 1.5483795404434204,
590
+ "logits/rejected": 1.4731425046920776,
591
+ "logps/chosen": -161.77774047851562,
592
+ "logps/rejected": -168.05458068847656,
593
+ "loss": 0.6911417007446289,
594
+ "rewards/accuracies": 0.550000011920929,
595
+ "rewards/chosen": 0.050054989755153656,
596
+ "rewards/margins": 0.0043535237200558186,
597
+ "rewards/rejected": 0.045701466500759125,
598
+ "step": 360
599
+ },
600
+ {
601
+ "epoch": 0.296,
602
+ "grad_norm": 4.362200736999512,
603
+ "learning_rate": 9.471111111111111e-07,
604
+ "logits/chosen": 1.735099196434021,
605
+ "logits/rejected": 1.7567729949951172,
606
+ "logps/chosen": -161.30374145507812,
607
+ "logps/rejected": -153.4731903076172,
608
+ "loss": 0.6882652282714844,
609
+ "rewards/accuracies": 0.5625,
610
+ "rewards/chosen": 0.06078929826617241,
611
+ "rewards/margins": 0.010189466178417206,
612
+ "rewards/rejected": 0.0505998320877552,
613
+ "step": 370
614
+ },
615
+ {
616
+ "epoch": 0.304,
617
+ "grad_norm": 4.35581111907959,
618
+ "learning_rate": 9.426666666666666e-07,
619
+ "logits/chosen": 1.604020118713379,
620
+ "logits/rejected": 1.524717926979065,
621
+ "logps/chosen": -141.42324829101562,
622
+ "logps/rejected": -151.82521057128906,
623
+ "loss": 0.6888412952423095,
624
+ "rewards/accuracies": 0.5375000238418579,
625
+ "rewards/chosen": 0.06947806477546692,
626
+ "rewards/margins": 0.009350637905299664,
627
+ "rewards/rejected": 0.06012742593884468,
628
+ "step": 380
629
+ },
630
+ {
631
+ "epoch": 0.312,
632
+ "grad_norm": 4.360926628112793,
633
+ "learning_rate": 9.382222222222222e-07,
634
+ "logits/chosen": 1.7075706720352173,
635
+ "logits/rejected": 1.6819493770599365,
636
+ "logps/chosen": -150.63375854492188,
637
+ "logps/rejected": -137.05673217773438,
638
+ "loss": 0.6836989879608154,
639
+ "rewards/accuracies": 0.612500011920929,
640
+ "rewards/chosen": 0.08739721775054932,
641
+ "rewards/margins": 0.019574418663978577,
642
+ "rewards/rejected": 0.06782279908657074,
643
+ "step": 390
644
+ },
645
+ {
646
+ "epoch": 0.32,
647
+ "grad_norm": 4.909813404083252,
648
+ "learning_rate": 9.337777777777778e-07,
649
+ "logits/chosen": 1.6920162439346313,
650
+ "logits/rejected": 1.675100326538086,
651
+ "logps/chosen": -154.39663696289062,
652
+ "logps/rejected": -147.51455688476562,
653
+ "loss": 0.6892420768737793,
654
+ "rewards/accuracies": 0.5,
655
+ "rewards/chosen": 0.08908190578222275,
656
+ "rewards/margins": 0.008944114670157433,
657
+ "rewards/rejected": 0.08013778924942017,
658
+ "step": 400
659
+ },
660
+ {
661
+ "epoch": 0.32,
662
+ "eval_logits/chosen": 1.5610027313232422,
663
+ "eval_logits/rejected": 1.5909091234207153,
664
+ "eval_logps/chosen": -152.39871215820312,
665
+ "eval_logps/rejected": -147.04049682617188,
666
+ "eval_loss": 0.6903401017189026,
667
+ "eval_rewards/accuracies": 0.5400000214576721,
668
+ "eval_rewards/chosen": 0.09304190427064896,
669
+ "eval_rewards/margins": 0.006809028796851635,
670
+ "eval_rewards/rejected": 0.08623287081718445,
671
+ "eval_runtime": 90.296,
672
+ "eval_samples_per_second": 5.537,
673
+ "eval_steps_per_second": 2.769,
674
+ "step": 400
675
+ },
676
+ {
677
+ "epoch": 0.328,
678
+ "grad_norm": 4.541158199310303,
679
+ "learning_rate": 9.293333333333333e-07,
680
+ "logits/chosen": 1.6515496969223022,
681
+ "logits/rejected": 1.548688530921936,
682
+ "logps/chosen": -147.21546936035156,
683
+ "logps/rejected": -187.50816345214844,
684
+ "loss": 0.6924624919891358,
685
+ "rewards/accuracies": 0.512499988079071,
686
+ "rewards/chosen": 0.10388608276844025,
687
+ "rewards/margins": 0.002456969814375043,
688
+ "rewards/rejected": 0.10142910480499268,
689
+ "step": 410
690
+ },
691
+ {
692
+ "epoch": 0.336,
693
+ "grad_norm": 5.246954917907715,
694
+ "learning_rate": 9.248888888888888e-07,
695
+ "logits/chosen": 1.6460405588150024,
696
+ "logits/rejected": 1.6713184118270874,
697
+ "logps/chosen": -151.11341857910156,
698
+ "logps/rejected": -166.2979736328125,
699
+ "loss": 0.6986268043518067,
700
+ "rewards/accuracies": 0.4000000059604645,
701
+ "rewards/chosen": 0.09254096448421478,
702
+ "rewards/margins": -0.009933066554367542,
703
+ "rewards/rejected": 0.10247401893138885,
704
+ "step": 420
705
+ },
706
+ {
707
+ "epoch": 0.344,
708
+ "grad_norm": 4.068811416625977,
709
+ "learning_rate": 9.204444444444443e-07,
710
+ "logits/chosen": 1.6973702907562256,
711
+ "logits/rejected": 1.7605253458023071,
712
+ "logps/chosen": -162.7523651123047,
713
+ "logps/rejected": -150.79718017578125,
714
+ "loss": 0.6896752834320068,
715
+ "rewards/accuracies": 0.4749999940395355,
716
+ "rewards/chosen": 0.08682320266962051,
717
+ "rewards/margins": 0.007874277420341969,
718
+ "rewards/rejected": 0.07894892990589142,
719
+ "step": 430
720
+ },
721
+ {
722
+ "epoch": 0.352,
723
+ "grad_norm": 4.387909412384033,
724
+ "learning_rate": 9.16e-07,
725
+ "logits/chosen": 1.5456931591033936,
726
+ "logits/rejected": 1.4381892681121826,
727
+ "logps/chosen": -155.777099609375,
728
+ "logps/rejected": -144.95742797851562,
729
+ "loss": 0.6881390571594238,
730
+ "rewards/accuracies": 0.574999988079071,
731
+ "rewards/chosen": 0.08465877920389175,
732
+ "rewards/margins": 0.01117948442697525,
733
+ "rewards/rejected": 0.0734792947769165,
734
+ "step": 440
735
+ },
736
+ {
737
+ "epoch": 0.36,
738
+ "grad_norm": 4.3955864906311035,
739
+ "learning_rate": 9.115555555555555e-07,
740
+ "logits/chosen": 1.7298389673233032,
741
+ "logits/rejected": 1.681171178817749,
742
+ "logps/chosen": -156.2227783203125,
743
+ "logps/rejected": -158.81114196777344,
744
+ "loss": 0.685992956161499,
745
+ "rewards/accuracies": 0.574999988079071,
746
+ "rewards/chosen": 0.08095243573188782,
747
+ "rewards/margins": 0.015426402911543846,
748
+ "rewards/rejected": 0.06552603840827942,
749
+ "step": 450
750
+ },
751
+ {
752
+ "epoch": 0.368,
753
+ "grad_norm": 4.6138176918029785,
754
+ "learning_rate": 9.071111111111111e-07,
755
+ "logits/chosen": 1.6315510272979736,
756
+ "logits/rejected": 1.6908462047576904,
757
+ "logps/chosen": -150.84512329101562,
758
+ "logps/rejected": -163.89492797851562,
759
+ "loss": 0.6891673088073731,
760
+ "rewards/accuracies": 0.550000011920929,
761
+ "rewards/chosen": 0.07254813611507416,
762
+ "rewards/margins": 0.008895034901797771,
763
+ "rewards/rejected": 0.06365309655666351,
764
+ "step": 460
765
+ },
766
+ {
767
+ "epoch": 0.376,
768
+ "grad_norm": 4.3172101974487305,
769
+ "learning_rate": 9.026666666666665e-07,
770
+ "logits/chosen": 1.4395225048065186,
771
+ "logits/rejected": 1.4489599466323853,
772
+ "logps/chosen": -130.1565399169922,
773
+ "logps/rejected": -122.24504089355469,
774
+ "loss": 0.6887143135070801,
775
+ "rewards/accuracies": 0.5249999761581421,
776
+ "rewards/chosen": 0.08355607837438583,
777
+ "rewards/margins": 0.009718736633658409,
778
+ "rewards/rejected": 0.07383735477924347,
779
+ "step": 470
780
+ },
781
+ {
782
+ "epoch": 0.384,
783
+ "grad_norm": 4.2122087478637695,
784
+ "learning_rate": 8.982222222222222e-07,
785
+ "logits/chosen": 1.5334614515304565,
786
+ "logits/rejected": 1.5769469738006592,
787
+ "logps/chosen": -147.21896362304688,
788
+ "logps/rejected": -162.89804077148438,
789
+ "loss": 0.6849615573883057,
790
+ "rewards/accuracies": 0.637499988079071,
791
+ "rewards/chosen": 0.10068760812282562,
792
+ "rewards/margins": 0.017685385420918465,
793
+ "rewards/rejected": 0.0830022394657135,
794
+ "step": 480
795
+ },
796
+ {
797
+ "epoch": 0.392,
798
+ "grad_norm": 4.740354061126709,
799
+ "learning_rate": 8.937777777777777e-07,
800
+ "logits/chosen": 1.6524379253387451,
801
+ "logits/rejected": 1.7100518941879272,
802
+ "logps/chosen": -142.10653686523438,
803
+ "logps/rejected": -158.3316192626953,
804
+ "loss": 0.696216630935669,
805
+ "rewards/accuracies": 0.44999998807907104,
806
+ "rewards/chosen": 0.08680267632007599,
807
+ "rewards/margins": -0.004664557985961437,
808
+ "rewards/rejected": 0.0914672389626503,
809
+ "step": 490
810
+ },
811
+ {
812
+ "epoch": 0.4,
813
+ "grad_norm": 3.6374881267547607,
814
+ "learning_rate": 8.893333333333333e-07,
815
+ "logits/chosen": 1.518328309059143,
816
+ "logits/rejected": 1.6029644012451172,
817
+ "logps/chosen": -143.19154357910156,
818
+ "logps/rejected": -134.3892059326172,
819
+ "loss": 0.6908615589141845,
820
+ "rewards/accuracies": 0.512499988079071,
821
+ "rewards/chosen": 0.06735874712467194,
822
+ "rewards/margins": 0.005541653838008642,
823
+ "rewards/rejected": 0.06181709095835686,
824
+ "step": 500
825
+ },
826
+ {
827
+ "epoch": 0.4,
828
+ "eval_logits/chosen": 1.5418404340744019,
829
+ "eval_logits/rejected": 1.571341633796692,
830
+ "eval_logps/chosen": -152.5870819091797,
831
+ "eval_logps/rejected": -147.23146057128906,
832
+ "eval_loss": 0.6903930902481079,
833
+ "eval_rewards/accuracies": 0.5220000147819519,
834
+ "eval_rewards/chosen": 0.07420650124549866,
835
+ "eval_rewards/margins": 0.007066408637911081,
836
+ "eval_rewards/rejected": 0.06714009493589401,
837
+ "eval_runtime": 90.217,
838
+ "eval_samples_per_second": 5.542,
839
+ "eval_steps_per_second": 2.771,
840
+ "step": 500
841
+ },
842
+ {
843
+ "epoch": 0.408,
844
+ "grad_norm": 4.53076171875,
845
+ "learning_rate": 8.848888888888888e-07,
846
+ "logits/chosen": 1.6947540044784546,
847
+ "logits/rejected": 1.6306483745574951,
848
+ "logps/chosen": -130.33372497558594,
849
+ "logps/rejected": -139.05648803710938,
850
+ "loss": 0.6863756656646729,
851
+ "rewards/accuracies": 0.550000011920929,
852
+ "rewards/chosen": 0.08373989164829254,
853
+ "rewards/margins": 0.014761297032237053,
854
+ "rewards/rejected": 0.06897859275341034,
855
+ "step": 510
856
+ },
857
+ {
858
+ "epoch": 0.416,
859
+ "grad_norm": 5.064472675323486,
860
+ "learning_rate": 8.804444444444445e-07,
861
+ "logits/chosen": 1.795907974243164,
862
+ "logits/rejected": 1.6805435419082642,
863
+ "logps/chosen": -165.10183715820312,
864
+ "logps/rejected": -170.87112426757812,
865
+ "loss": 0.6915029525756836,
866
+ "rewards/accuracies": 0.574999988079071,
867
+ "rewards/chosen": 0.07381857931613922,
868
+ "rewards/margins": 0.00408085435628891,
869
+ "rewards/rejected": 0.06973771750926971,
870
+ "step": 520
871
+ },
872
+ {
873
+ "epoch": 0.424,
874
+ "grad_norm": 4.472287178039551,
875
+ "learning_rate": 8.76e-07,
876
+ "logits/chosen": 1.7226626873016357,
877
+ "logits/rejected": 1.6465301513671875,
878
+ "logps/chosen": -165.50076293945312,
879
+ "logps/rejected": -167.12991333007812,
880
+ "loss": 0.6784487724304199,
881
+ "rewards/accuracies": 0.637499988079071,
882
+ "rewards/chosen": 0.09387621283531189,
883
+ "rewards/margins": 0.031213903799653053,
884
+ "rewards/rejected": 0.06266231089830399,
885
+ "step": 530
886
+ },
887
+ {
888
+ "epoch": 0.432,
889
+ "grad_norm": 4.193634033203125,
890
+ "learning_rate": 8.715555555555554e-07,
891
+ "logits/chosen": 1.7823143005371094,
892
+ "logits/rejected": 1.7374283075332642,
893
+ "logps/chosen": -180.05233764648438,
894
+ "logps/rejected": -157.24835205078125,
895
+ "loss": 0.6891638278961182,
896
+ "rewards/accuracies": 0.5625,
897
+ "rewards/chosen": 0.10219593346118927,
898
+ "rewards/margins": 0.009201721288263798,
899
+ "rewards/rejected": 0.09299422055482864,
900
+ "step": 540
901
+ },
902
+ {
903
+ "epoch": 0.44,
904
+ "grad_norm": 5.877465724945068,
905
+ "learning_rate": 8.671111111111111e-07,
906
+ "logits/chosen": 1.6558294296264648,
907
+ "logits/rejected": 1.7549035549163818,
908
+ "logps/chosen": -149.97171020507812,
909
+ "logps/rejected": -166.52127075195312,
910
+ "loss": 0.6909477233886718,
911
+ "rewards/accuracies": 0.5625,
912
+ "rewards/chosen": 0.09389887005090714,
913
+ "rewards/margins": 0.00655100354924798,
914
+ "rewards/rejected": 0.0873478576540947,
915
+ "step": 550
916
+ },
917
+ {
918
+ "epoch": 0.448,
919
+ "grad_norm": 3.9154303073883057,
920
+ "learning_rate": 8.626666666666666e-07,
921
+ "logits/chosen": 1.7343839406967163,
922
+ "logits/rejected": 1.6256252527236938,
923
+ "logps/chosen": -153.2657470703125,
924
+ "logps/rejected": -137.84548950195312,
925
+ "loss": 0.6832056045532227,
926
+ "rewards/accuracies": 0.574999988079071,
927
+ "rewards/chosen": 0.08965932577848434,
928
+ "rewards/margins": 0.02167549543082714,
929
+ "rewards/rejected": 0.06798382848501205,
930
+ "step": 560
931
+ },
932
+ {
933
+ "epoch": 0.456,
934
+ "grad_norm": 3.569357395172119,
935
+ "learning_rate": 8.582222222222222e-07,
936
+ "logits/chosen": 1.6020238399505615,
937
+ "logits/rejected": 1.5468555688858032,
938
+ "logps/chosen": -156.9928741455078,
939
+ "logps/rejected": -150.9776153564453,
940
+ "loss": 0.6857921123504639,
941
+ "rewards/accuracies": 0.550000011920929,
942
+ "rewards/chosen": 0.07846538722515106,
943
+ "rewards/margins": 0.016974106431007385,
944
+ "rewards/rejected": 0.06149129942059517,
945
+ "step": 570
946
+ },
947
+ {
948
+ "epoch": 0.464,
949
+ "grad_norm": 5.710695266723633,
950
+ "learning_rate": 8.537777777777777e-07,
951
+ "logits/chosen": 1.4293699264526367,
952
+ "logits/rejected": 1.583032250404358,
953
+ "logps/chosen": -134.98165893554688,
954
+ "logps/rejected": -153.61439514160156,
955
+ "loss": 0.6899324417114258,
956
+ "rewards/accuracies": 0.5249999761581421,
957
+ "rewards/chosen": 0.05123991519212723,
958
+ "rewards/margins": 0.008615568280220032,
959
+ "rewards/rejected": 0.042624346911907196,
960
+ "step": 580
961
+ },
962
+ {
963
+ "epoch": 0.472,
964
+ "grad_norm": 3.757844924926758,
965
+ "learning_rate": 8.493333333333334e-07,
966
+ "logits/chosen": 1.5719819068908691,
967
+ "logits/rejected": 1.5706799030303955,
968
+ "logps/chosen": -143.9678955078125,
969
+ "logps/rejected": -130.64585876464844,
970
+ "loss": 0.6851204395294189,
971
+ "rewards/accuracies": 0.550000011920929,
972
+ "rewards/chosen": 0.07710663974285126,
973
+ "rewards/margins": 0.01796458289027214,
974
+ "rewards/rejected": 0.059142060577869415,
975
+ "step": 590
976
+ },
977
+ {
978
+ "epoch": 0.48,
979
+ "grad_norm": 5.662181854248047,
980
+ "learning_rate": 8.448888888888888e-07,
981
+ "logits/chosen": 1.6224644184112549,
982
+ "logits/rejected": 1.6623615026474,
983
+ "logps/chosen": -130.7429962158203,
984
+ "logps/rejected": -157.59295654296875,
985
+ "loss": 0.6958520889282227,
986
+ "rewards/accuracies": 0.5874999761581421,
987
+ "rewards/chosen": 0.0648646205663681,
988
+ "rewards/margins": -0.003241670085117221,
989
+ "rewards/rejected": 0.0681062787771225,
990
+ "step": 600
991
+ },
992
+ {
993
+ "epoch": 0.48,
994
+ "eval_logits/chosen": 1.5268914699554443,
995
+ "eval_logits/rejected": 1.5556302070617676,
996
+ "eval_logps/chosen": -152.5731201171875,
997
+ "eval_logps/rejected": -147.22303771972656,
998
+ "eval_loss": 0.6907246708869934,
999
+ "eval_rewards/accuracies": 0.5139999985694885,
1000
+ "eval_rewards/chosen": 0.07560181617736816,
1001
+ "eval_rewards/margins": 0.007620053365826607,
1002
+ "eval_rewards/rejected": 0.0679817646741867,
1003
+ "eval_runtime": 90.3327,
1004
+ "eval_samples_per_second": 5.535,
1005
+ "eval_steps_per_second": 2.768,
1006
+ "step": 600
1007
+ },
1008
+ {
1009
+ "epoch": 0.488,
1010
+ "grad_norm": 3.7953426837921143,
1011
+ "learning_rate": 8.404444444444444e-07,
1012
+ "logits/chosen": 1.6380192041397095,
1013
+ "logits/rejected": 1.6921494007110596,
1014
+ "logps/chosen": -130.59445190429688,
1015
+ "logps/rejected": -148.48709106445312,
1016
+ "loss": 0.6903901100158691,
1017
+ "rewards/accuracies": 0.5625,
1018
+ "rewards/chosen": 0.08007006347179413,
1019
+ "rewards/margins": 0.00720745325088501,
1020
+ "rewards/rejected": 0.07286261022090912,
1021
+ "step": 610
1022
+ },
1023
+ {
1024
+ "epoch": 0.496,
1025
+ "grad_norm": 7.121775150299072,
1026
+ "learning_rate": 8.359999999999999e-07,
1027
+ "logits/chosen": 1.6000845432281494,
1028
+ "logits/rejected": 1.731951355934143,
1029
+ "logps/chosen": -154.8905792236328,
1030
+ "logps/rejected": -166.4490966796875,
1031
+ "loss": 0.6969138145446777,
1032
+ "rewards/accuracies": 0.4749999940395355,
1033
+ "rewards/chosen": 0.05843223258852959,
1034
+ "rewards/margins": -0.0054575116373598576,
1035
+ "rewards/rejected": 0.06388974189758301,
1036
+ "step": 620
1037
+ },
1038
+ {
1039
+ "epoch": 0.504,
1040
+ "grad_norm": 5.155455589294434,
1041
+ "learning_rate": 8.315555555555556e-07,
1042
+ "logits/chosen": 1.6201622486114502,
1043
+ "logits/rejected": 1.6479911804199219,
1044
+ "logps/chosen": -165.98980712890625,
1045
+ "logps/rejected": -145.71644592285156,
1046
+ "loss": 0.6804090023040772,
1047
+ "rewards/accuracies": 0.612500011920929,
1048
+ "rewards/chosen": 0.0773121565580368,
1049
+ "rewards/margins": 0.027506589889526367,
1050
+ "rewards/rejected": 0.049805570393800735,
1051
+ "step": 630
1052
+ },
1053
+ {
1054
+ "epoch": 0.512,
1055
+ "grad_norm": 4.009693145751953,
1056
+ "learning_rate": 8.271111111111111e-07,
1057
+ "logits/chosen": 1.5530269145965576,
1058
+ "logits/rejected": 1.5585509538650513,
1059
+ "logps/chosen": -166.77560424804688,
1060
+ "logps/rejected": -151.09249877929688,
1061
+ "loss": 0.6879617691040039,
1062
+ "rewards/accuracies": 0.512499988079071,
1063
+ "rewards/chosen": 0.07863648235797882,
1064
+ "rewards/margins": 0.014577758498489857,
1065
+ "rewards/rejected": 0.06405872106552124,
1066
+ "step": 640
1067
+ },
1068
+ {
1069
+ "epoch": 0.52,
1070
+ "grad_norm": 4.472072601318359,
1071
+ "learning_rate": 8.226666666666666e-07,
1072
+ "logits/chosen": 1.662239670753479,
1073
+ "logits/rejected": 1.6585584878921509,
1074
+ "logps/chosen": -153.26776123046875,
1075
+ "logps/rejected": -125.24166107177734,
1076
+ "loss": 0.6882720470428467,
1077
+ "rewards/accuracies": 0.5,
1078
+ "rewards/chosen": 0.09423185139894485,
1079
+ "rewards/margins": 0.011949598789215088,
1080
+ "rewards/rejected": 0.08228223770856857,
1081
+ "step": 650
1082
+ },
1083
+ {
1084
+ "epoch": 0.528,
1085
+ "grad_norm": 5.073488712310791,
1086
+ "learning_rate": 8.182222222222222e-07,
1087
+ "logits/chosen": 1.6752973794937134,
1088
+ "logits/rejected": 1.6020495891571045,
1089
+ "logps/chosen": -150.0669708251953,
1090
+ "logps/rejected": -131.1305694580078,
1091
+ "loss": 0.6880992889404297,
1092
+ "rewards/accuracies": 0.6000000238418579,
1093
+ "rewards/chosen": 0.07629784196615219,
1094
+ "rewards/margins": 0.01174530852586031,
1095
+ "rewards/rejected": 0.06455253064632416,
1096
+ "step": 660
1097
+ },
1098
+ {
1099
+ "epoch": 0.536,
1100
+ "grad_norm": 5.938063621520996,
1101
+ "learning_rate": 8.137777777777777e-07,
1102
+ "logits/chosen": 1.7563416957855225,
1103
+ "logits/rejected": 1.5739262104034424,
1104
+ "logps/chosen": -165.046875,
1105
+ "logps/rejected": -150.13104248046875,
1106
+ "loss": 0.6939912796020508,
1107
+ "rewards/accuracies": 0.4625000059604645,
1108
+ "rewards/chosen": 0.09020708501338959,
1109
+ "rewards/margins": 0.0006308574229478836,
1110
+ "rewards/rejected": 0.08957622945308685,
1111
+ "step": 670
1112
+ },
1113
+ {
1114
+ "epoch": 0.544,
1115
+ "grad_norm": 4.362247467041016,
1116
+ "learning_rate": 8.093333333333333e-07,
1117
+ "logits/chosen": 1.6460363864898682,
1118
+ "logits/rejected": 1.6379966735839844,
1119
+ "logps/chosen": -143.24754333496094,
1120
+ "logps/rejected": -131.8529815673828,
1121
+ "loss": 0.6835652351379394,
1122
+ "rewards/accuracies": 0.5874999761581421,
1123
+ "rewards/chosen": 0.10346712917089462,
1124
+ "rewards/margins": 0.02079077437520027,
1125
+ "rewards/rejected": 0.08267635107040405,
1126
+ "step": 680
1127
+ },
1128
+ {
1129
+ "epoch": 0.552,
1130
+ "grad_norm": 4.323369026184082,
1131
+ "learning_rate": 8.048888888888888e-07,
1132
+ "logits/chosen": 1.5466216802597046,
1133
+ "logits/rejected": 1.541775107383728,
1134
+ "logps/chosen": -171.82138061523438,
1135
+ "logps/rejected": -158.87603759765625,
1136
+ "loss": 0.6892048358917237,
1137
+ "rewards/accuracies": 0.5375000238418579,
1138
+ "rewards/chosen": 0.0915951356291771,
1139
+ "rewards/margins": 0.009338131174445152,
1140
+ "rewards/rejected": 0.08225701004266739,
1141
+ "step": 690
1142
+ },
1143
+ {
1144
+ "epoch": 0.56,
1145
+ "grad_norm": 5.28114128112793,
1146
+ "learning_rate": 8.004444444444444e-07,
1147
+ "logits/chosen": 1.6419496536254883,
1148
+ "logits/rejected": 1.6641361713409424,
1149
+ "logps/chosen": -158.20787048339844,
1150
+ "logps/rejected": -136.3108367919922,
1151
+ "loss": 0.6985964775085449,
1152
+ "rewards/accuracies": 0.4625000059604645,
1153
+ "rewards/chosen": 0.0700095146894455,
1154
+ "rewards/margins": -0.008601363748311996,
1155
+ "rewards/rejected": 0.07861088216304779,
1156
+ "step": 700
1157
+ },
1158
+ {
1159
+ "epoch": 0.56,
1160
+ "eval_logits/chosen": 1.5386524200439453,
1161
+ "eval_logits/rejected": 1.5675796270370483,
1162
+ "eval_logps/chosen": -152.39877319335938,
1163
+ "eval_logps/rejected": -147.0768585205078,
1164
+ "eval_loss": 0.6893304586410522,
1165
+ "eval_rewards/accuracies": 0.5320000052452087,
1166
+ "eval_rewards/chosen": 0.09303704649209976,
1167
+ "eval_rewards/margins": 0.010439171455800533,
1168
+ "eval_rewards/rejected": 0.08259786665439606,
1169
+ "eval_runtime": 90.3103,
1170
+ "eval_samples_per_second": 5.536,
1171
+ "eval_steps_per_second": 2.768,
1172
+ "step": 700
1173
+ },
1174
+ {
1175
+ "epoch": 0.568,
1176
+ "grad_norm": 4.867155075073242,
1177
+ "learning_rate": 7.96e-07,
1178
+ "logits/chosen": 1.7137393951416016,
1179
+ "logits/rejected": 1.6643224954605103,
1180
+ "logps/chosen": -147.054931640625,
1181
+ "logps/rejected": -162.10067749023438,
1182
+ "loss": 0.6891860008239746,
1183
+ "rewards/accuracies": 0.6000000238418579,
1184
+ "rewards/chosen": 0.11649386584758759,
1185
+ "rewards/margins": 0.00985223613679409,
1186
+ "rewards/rejected": 0.10664163529872894,
1187
+ "step": 710
1188
+ },
1189
+ {
1190
+ "epoch": 0.576,
1191
+ "grad_norm": 4.687198638916016,
1192
+ "learning_rate": 7.915555555555556e-07,
1193
+ "logits/chosen": 1.686532974243164,
1194
+ "logits/rejected": 1.7992823123931885,
1195
+ "logps/chosen": -138.60238647460938,
1196
+ "logps/rejected": -134.22702026367188,
1197
+ "loss": 0.7006660461425781,
1198
+ "rewards/accuracies": 0.4625000059604645,
1199
+ "rewards/chosen": 0.08661777526140213,
1200
+ "rewards/margins": -0.013613695278763771,
1201
+ "rewards/rejected": 0.10023146867752075,
1202
+ "step": 720
1203
+ },
1204
+ {
1205
+ "epoch": 0.584,
1206
+ "grad_norm": 4.63344669342041,
1207
+ "learning_rate": 7.87111111111111e-07,
1208
+ "logits/chosen": 1.781561255455017,
1209
+ "logits/rejected": 1.7561432123184204,
1210
+ "logps/chosen": -151.60018920898438,
1211
+ "logps/rejected": -147.93264770507812,
1212
+ "loss": 0.6958267688751221,
1213
+ "rewards/accuracies": 0.5,
1214
+ "rewards/chosen": 0.10642895847558975,
1215
+ "rewards/margins": -0.002830044599249959,
1216
+ "rewards/rejected": 0.1092589944601059,
1217
+ "step": 730
1218
+ },
1219
+ {
1220
+ "epoch": 0.592,
1221
+ "grad_norm": 4.5400800704956055,
1222
+ "learning_rate": 7.826666666666666e-07,
1223
+ "logits/chosen": 1.670771837234497,
1224
+ "logits/rejected": 1.5866410732269287,
1225
+ "logps/chosen": -155.36764526367188,
1226
+ "logps/rejected": -132.60902404785156,
1227
+ "loss": 0.6921723842620849,
1228
+ "rewards/accuracies": 0.5375000238418579,
1229
+ "rewards/chosen": 0.0910472720861435,
1230
+ "rewards/margins": 0.003630922408774495,
1231
+ "rewards/rejected": 0.08741635084152222,
1232
+ "step": 740
1233
+ },
1234
+ {
1235
+ "epoch": 0.6,
1236
+ "grad_norm": 4.779706954956055,
1237
+ "learning_rate": 7.782222222222222e-07,
1238
+ "logits/chosen": 1.531534194946289,
1239
+ "logits/rejected": 1.5548356771469116,
1240
+ "logps/chosen": -135.88177490234375,
1241
+ "logps/rejected": -157.09231567382812,
1242
+ "loss": 0.6919455528259277,
1243
+ "rewards/accuracies": 0.5375000238418579,
1244
+ "rewards/chosen": 0.09000497311353683,
1245
+ "rewards/margins": 0.004116452299058437,
1246
+ "rewards/rejected": 0.08588851988315582,
1247
+ "step": 750
1248
+ },
1249
+ {
1250
+ "epoch": 0.608,
1251
+ "grad_norm": 5.283969879150391,
1252
+ "learning_rate": 7.737777777777777e-07,
1253
+ "logits/chosen": 1.6809686422348022,
1254
+ "logits/rejected": 1.501511812210083,
1255
+ "logps/chosen": -137.67315673828125,
1256
+ "logps/rejected": -128.26022338867188,
1257
+ "loss": 0.6907838344573974,
1258
+ "rewards/accuracies": 0.5375000238418579,
1259
+ "rewards/chosen": 0.08058954030275345,
1260
+ "rewards/margins": 0.0064643076620996,
1261
+ "rewards/rejected": 0.07412523031234741,
1262
+ "step": 760
1263
+ },
1264
+ {
1265
+ "epoch": 0.616,
1266
+ "grad_norm": 4.341912269592285,
1267
+ "learning_rate": 7.693333333333333e-07,
1268
+ "logits/chosen": 1.684203863143921,
1269
+ "logits/rejected": 1.6489808559417725,
1270
+ "logps/chosen": -139.82455444335938,
1271
+ "logps/rejected": -135.16998291015625,
1272
+ "loss": 0.6793179988861084,
1273
+ "rewards/accuracies": 0.6499999761581421,
1274
+ "rewards/chosen": 0.10346569865942001,
1275
+ "rewards/margins": 0.02992106042802334,
1276
+ "rewards/rejected": 0.07354463636875153,
1277
+ "step": 770
1278
+ },
1279
+ {
1280
+ "epoch": 0.624,
1281
+ "grad_norm": 5.209469318389893,
1282
+ "learning_rate": 7.648888888888888e-07,
1283
+ "logits/chosen": 1.5599935054779053,
1284
+ "logits/rejected": 1.6487398147583008,
1285
+ "logps/chosen": -152.46170043945312,
1286
+ "logps/rejected": -157.7329559326172,
1287
+ "loss": 0.6873491287231446,
1288
+ "rewards/accuracies": 0.5625,
1289
+ "rewards/chosen": 0.10296590626239777,
1290
+ "rewards/margins": 0.01368915755301714,
1291
+ "rewards/rejected": 0.0892767459154129,
1292
+ "step": 780
1293
+ },
1294
+ {
1295
+ "epoch": 0.632,
1296
+ "grad_norm": 5.364309310913086,
1297
+ "learning_rate": 7.604444444444445e-07,
1298
+ "logits/chosen": 1.5357733964920044,
1299
+ "logits/rejected": 1.5833505392074585,
1300
+ "logps/chosen": -146.4203338623047,
1301
+ "logps/rejected": -149.77499389648438,
1302
+ "loss": 0.68800368309021,
1303
+ "rewards/accuracies": 0.5375000238418579,
1304
+ "rewards/chosen": 0.0752335637807846,
1305
+ "rewards/margins": 0.01265893317759037,
1306
+ "rewards/rejected": 0.06257463991641998,
1307
+ "step": 790
1308
+ },
1309
+ {
1310
+ "epoch": 0.64,
1311
+ "grad_norm": 4.701781272888184,
1312
+ "learning_rate": 7.559999999999999e-07,
1313
+ "logits/chosen": 1.7693378925323486,
1314
+ "logits/rejected": 1.784106969833374,
1315
+ "logps/chosen": -178.75717163085938,
1316
+ "logps/rejected": -192.69229125976562,
1317
+ "loss": 0.7001357078552246,
1318
+ "rewards/accuracies": 0.4625000059604645,
1319
+ "rewards/chosen": 0.07940518856048584,
1320
+ "rewards/margins": -0.01138945110142231,
1321
+ "rewards/rejected": 0.0907946228981018,
1322
+ "step": 800
1323
+ },
1324
+ {
1325
+ "epoch": 0.64,
1326
+ "eval_logits/chosen": 1.532821536064148,
1327
+ "eval_logits/rejected": 1.5615730285644531,
1328
+ "eval_logps/chosen": -152.57040405273438,
1329
+ "eval_logps/rejected": -147.24534606933594,
1330
+ "eval_loss": 0.6894002556800842,
1331
+ "eval_rewards/accuracies": 0.5339999794960022,
1332
+ "eval_rewards/chosen": 0.07587439566850662,
1333
+ "eval_rewards/margins": 0.010124183259904385,
1334
+ "eval_rewards/rejected": 0.06575021147727966,
1335
+ "eval_runtime": 90.2864,
1336
+ "eval_samples_per_second": 5.538,
1337
+ "eval_steps_per_second": 2.769,
1338
+ "step": 800
1339
+ },
1340
+ {
1341
+ "epoch": 0.648,
1342
+ "grad_norm": 4.731827259063721,
1343
+ "learning_rate": 7.515555555555555e-07,
1344
+ "logits/chosen": 1.5014355182647705,
1345
+ "logits/rejected": 1.706011176109314,
1346
+ "logps/chosen": -113.23974609375,
1347
+ "logps/rejected": -150.55316162109375,
1348
+ "loss": 0.6896101951599121,
1349
+ "rewards/accuracies": 0.550000011920929,
1350
+ "rewards/chosen": 0.066777303814888,
1351
+ "rewards/margins": 0.008613836951553822,
1352
+ "rewards/rejected": 0.05816347524523735,
1353
+ "step": 810
1354
+ },
1355
+ {
1356
+ "epoch": 0.656,
1357
+ "grad_norm": 4.050163745880127,
1358
+ "learning_rate": 7.47111111111111e-07,
1359
+ "logits/chosen": 1.701898217201233,
1360
+ "logits/rejected": 1.7274971008300781,
1361
+ "logps/chosen": -147.45330810546875,
1362
+ "logps/rejected": -140.33255004882812,
1363
+ "loss": 0.6757836818695069,
1364
+ "rewards/accuracies": 0.6875,
1365
+ "rewards/chosen": 0.08680602163076401,
1366
+ "rewards/margins": 0.03675536438822746,
1367
+ "rewards/rejected": 0.050050657242536545,
1368
+ "step": 820
1369
+ },
1370
+ {
1371
+ "epoch": 0.664,
1372
+ "grad_norm": 4.168673992156982,
1373
+ "learning_rate": 7.426666666666667e-07,
1374
+ "logits/chosen": 1.6135514974594116,
1375
+ "logits/rejected": 1.6518815755844116,
1376
+ "logps/chosen": -137.38467407226562,
1377
+ "logps/rejected": -132.65890502929688,
1378
+ "loss": 0.6800778865814209,
1379
+ "rewards/accuracies": 0.637499988079071,
1380
+ "rewards/chosen": 0.07235284894704819,
1381
+ "rewards/margins": 0.028275374323129654,
1382
+ "rewards/rejected": 0.04407747834920883,
1383
+ "step": 830
1384
+ },
1385
+ {
1386
+ "epoch": 0.672,
1387
+ "grad_norm": 4.72458028793335,
1388
+ "learning_rate": 7.382222222222222e-07,
1389
+ "logits/chosen": 1.5987484455108643,
1390
+ "logits/rejected": 1.6328668594360352,
1391
+ "logps/chosen": -146.712158203125,
1392
+ "logps/rejected": -156.0950469970703,
1393
+ "loss": 0.6804145336151123,
1394
+ "rewards/accuracies": 0.5625,
1395
+ "rewards/chosen": 0.07264034450054169,
1396
+ "rewards/margins": 0.0286283977329731,
1397
+ "rewards/rejected": 0.04401194304227829,
1398
+ "step": 840
1399
+ },
1400
+ {
1401
+ "epoch": 0.68,
1402
+ "grad_norm": 4.735199928283691,
1403
+ "learning_rate": 7.337777777777778e-07,
1404
+ "logits/chosen": 1.6810247898101807,
1405
+ "logits/rejected": 1.6662237644195557,
1406
+ "logps/chosen": -159.40650939941406,
1407
+ "logps/rejected": -140.65591430664062,
1408
+ "loss": 0.6805107116699218,
1409
+ "rewards/accuracies": 0.574999988079071,
1410
+ "rewards/chosen": 0.061445970088243484,
1411
+ "rewards/margins": 0.02743927761912346,
1412
+ "rewards/rejected": 0.034006692469120026,
1413
+ "step": 850
1414
+ },
1415
+ {
1416
+ "epoch": 0.688,
1417
+ "grad_norm": 3.7038252353668213,
1418
+ "learning_rate": 7.293333333333332e-07,
1419
+ "logits/chosen": 1.6597106456756592,
1420
+ "logits/rejected": 1.6951271295547485,
1421
+ "logps/chosen": -138.1852569580078,
1422
+ "logps/rejected": -128.6427764892578,
1423
+ "loss": 0.6821750164031982,
1424
+ "rewards/accuracies": 0.5375000238418579,
1425
+ "rewards/chosen": 0.07160943746566772,
1426
+ "rewards/margins": 0.023774990811944008,
1427
+ "rewards/rejected": 0.047834448516368866,
1428
+ "step": 860
1429
+ },
1430
+ {
1431
+ "epoch": 0.696,
1432
+ "grad_norm": 4.820807456970215,
1433
+ "learning_rate": 7.248888888888888e-07,
1434
+ "logits/chosen": 1.5708585977554321,
1435
+ "logits/rejected": 1.5483477115631104,
1436
+ "logps/chosen": -152.8867950439453,
1437
+ "logps/rejected": -152.02584838867188,
1438
+ "loss": 0.6911486625671387,
1439
+ "rewards/accuracies": 0.550000011920929,
1440
+ "rewards/chosen": 0.04338831081986427,
1441
+ "rewards/margins": 0.006898392923176289,
1442
+ "rewards/rejected": 0.03648992255330086,
1443
+ "step": 870
1444
+ },
1445
+ {
1446
+ "epoch": 0.704,
1447
+ "grad_norm": 4.6849493980407715,
1448
+ "learning_rate": 7.204444444444444e-07,
1449
+ "logits/chosen": 1.5262442827224731,
1450
+ "logits/rejected": 1.7751166820526123,
1451
+ "logps/chosen": -143.77993774414062,
1452
+ "logps/rejected": -155.7498016357422,
1453
+ "loss": 0.6910871028900146,
1454
+ "rewards/accuracies": 0.5249999761581421,
1455
+ "rewards/chosen": 0.061417657881975174,
1456
+ "rewards/margins": 0.006087464280426502,
1457
+ "rewards/rejected": 0.055330194532871246,
1458
+ "step": 880
1459
+ },
1460
+ {
1461
+ "epoch": 0.712,
1462
+ "grad_norm": 4.951540946960449,
1463
+ "learning_rate": 7.159999999999999e-07,
1464
+ "logits/chosen": 1.405790090560913,
1465
+ "logits/rejected": 1.5980100631713867,
1466
+ "logps/chosen": -147.6872100830078,
1467
+ "logps/rejected": -160.23947143554688,
1468
+ "loss": 0.6822467803955078,
1469
+ "rewards/accuracies": 0.5874999761581421,
1470
+ "rewards/chosen": 0.049768321216106415,
1471
+ "rewards/margins": 0.024227874353528023,
1472
+ "rewards/rejected": 0.02554045058786869,
1473
+ "step": 890
1474
+ },
1475
+ {
1476
+ "epoch": 0.72,
1477
+ "grad_norm": 4.255526542663574,
1478
+ "learning_rate": 7.115555555555556e-07,
1479
+ "logits/chosen": 1.6527436971664429,
1480
+ "logits/rejected": 1.787755012512207,
1481
+ "logps/chosen": -164.73355102539062,
1482
+ "logps/rejected": -178.98507690429688,
1483
+ "loss": 0.6861439704895019,
1484
+ "rewards/accuracies": 0.5,
1485
+ "rewards/chosen": 0.07546674460172653,
1486
+ "rewards/margins": 0.017860155552625656,
1487
+ "rewards/rejected": 0.05760659649968147,
1488
+ "step": 900
1489
+ },
1490
+ {
1491
+ "epoch": 0.72,
1492
+ "eval_logits/chosen": 1.5091361999511719,
1493
+ "eval_logits/rejected": 1.5371856689453125,
1494
+ "eval_logps/chosen": -152.8962860107422,
1495
+ "eval_logps/rejected": -147.56655883789062,
1496
+ "eval_loss": 0.690089225769043,
1497
+ "eval_rewards/accuracies": 0.527999997138977,
1498
+ "eval_rewards/chosen": 0.04328843951225281,
1499
+ "eval_rewards/margins": 0.009660834446549416,
1500
+ "eval_rewards/rejected": 0.03362761065363884,
1501
+ "eval_runtime": 90.3227,
1502
+ "eval_samples_per_second": 5.536,
1503
+ "eval_steps_per_second": 2.768,
1504
+ "step": 900
1505
+ },
1506
+ {
1507
+ "epoch": 0.728,
1508
+ "grad_norm": 4.453512668609619,
1509
+ "learning_rate": 7.071111111111111e-07,
1510
+ "logits/chosen": 1.700484037399292,
1511
+ "logits/rejected": 1.4941186904907227,
1512
+ "logps/chosen": -138.50682067871094,
1513
+ "logps/rejected": -137.02490234375,
1514
+ "loss": 0.6877517700195312,
1515
+ "rewards/accuracies": 0.574999988079071,
1516
+ "rewards/chosen": 0.05909284949302673,
1517
+ "rewards/margins": 0.013610092923045158,
1518
+ "rewards/rejected": 0.04548276215791702,
1519
+ "step": 910
1520
+ },
1521
+ {
1522
+ "epoch": 0.736,
1523
+ "grad_norm": 5.548420429229736,
1524
+ "learning_rate": 7.026666666666667e-07,
1525
+ "logits/chosen": 1.409182071685791,
1526
+ "logits/rejected": 1.375797152519226,
1527
+ "logps/chosen": -158.7325439453125,
1528
+ "logps/rejected": -161.9824981689453,
1529
+ "loss": 0.6867617607116699,
1530
+ "rewards/accuracies": 0.5249999761581421,
1531
+ "rewards/chosen": 0.07141564786434174,
1532
+ "rewards/margins": 0.015245514921844006,
1533
+ "rewards/rejected": 0.05617012828588486,
1534
+ "step": 920
1535
+ },
1536
+ {
1537
+ "epoch": 0.744,
1538
+ "grad_norm": 5.186211109161377,
1539
+ "learning_rate": 6.982222222222221e-07,
1540
+ "logits/chosen": 1.6255321502685547,
1541
+ "logits/rejected": 1.7182047367095947,
1542
+ "logps/chosen": -156.53213500976562,
1543
+ "logps/rejected": -160.67556762695312,
1544
+ "loss": 0.6812005519866944,
1545
+ "rewards/accuracies": 0.574999988079071,
1546
+ "rewards/chosen": 0.062074560672044754,
1547
+ "rewards/margins": 0.026883777230978012,
1548
+ "rewards/rejected": 0.03519078344106674,
1549
+ "step": 930
1550
+ },
1551
+ {
1552
+ "epoch": 0.752,
1553
+ "grad_norm": 4.044335842132568,
1554
+ "learning_rate": 6.937777777777778e-07,
1555
+ "logits/chosen": 1.6656415462493896,
1556
+ "logits/rejected": 1.7865594625473022,
1557
+ "logps/chosen": -147.00344848632812,
1558
+ "logps/rejected": -173.11428833007812,
1559
+ "loss": 0.690484619140625,
1560
+ "rewards/accuracies": 0.5375000238418579,
1561
+ "rewards/chosen": 0.0269255880266428,
1562
+ "rewards/margins": 0.007569611072540283,
1563
+ "rewards/rejected": 0.019355975091457367,
1564
+ "step": 940
1565
+ },
1566
+ {
1567
+ "epoch": 0.76,
1568
+ "grad_norm": 4.8925299644470215,
1569
+ "learning_rate": 6.893333333333333e-07,
1570
+ "logits/chosen": 1.4856427907943726,
1571
+ "logits/rejected": 1.5664136409759521,
1572
+ "logps/chosen": -139.77938842773438,
1573
+ "logps/rejected": -152.74557495117188,
1574
+ "loss": 0.6817938804626464,
1575
+ "rewards/accuracies": 0.574999988079071,
1576
+ "rewards/chosen": 0.05857279896736145,
1577
+ "rewards/margins": 0.02614629827439785,
1578
+ "rewards/rejected": 0.03242649883031845,
1579
+ "step": 950
1580
+ },
1581
+ {
1582
+ "epoch": 0.768,
1583
+ "grad_norm": 4.514585018157959,
1584
+ "learning_rate": 6.848888888888889e-07,
1585
+ "logits/chosen": 1.6077378988265991,
1586
+ "logits/rejected": 1.4770267009735107,
1587
+ "logps/chosen": -134.602294921875,
1588
+ "logps/rejected": -117.98567199707031,
1589
+ "loss": 0.694350004196167,
1590
+ "rewards/accuracies": 0.44999998807907104,
1591
+ "rewards/chosen": 0.028881916776299477,
1592
+ "rewards/margins": 0.00047348294174298644,
1593
+ "rewards/rejected": 0.028408434242010117,
1594
+ "step": 960
1595
+ },
1596
+ {
1597
+ "epoch": 0.776,
1598
+ "grad_norm": 3.9295125007629395,
1599
+ "learning_rate": 6.804444444444444e-07,
1600
+ "logits/chosen": 1.7124595642089844,
1601
+ "logits/rejected": 1.8135782480239868,
1602
+ "logps/chosen": -152.15426635742188,
1603
+ "logps/rejected": -147.62945556640625,
1604
+ "loss": 0.6929523944854736,
1605
+ "rewards/accuracies": 0.5375000238418579,
1606
+ "rewards/chosen": 0.057992029935121536,
1607
+ "rewards/margins": 0.002844708738848567,
1608
+ "rewards/rejected": 0.05514732003211975,
1609
+ "step": 970
1610
+ },
1611
+ {
1612
+ "epoch": 0.784,
1613
+ "grad_norm": 5.510717391967773,
1614
+ "learning_rate": 6.76e-07,
1615
+ "logits/chosen": 1.483984351158142,
1616
+ "logits/rejected": 1.4224226474761963,
1617
+ "logps/chosen": -167.27133178710938,
1618
+ "logps/rejected": -140.16891479492188,
1619
+ "loss": 0.6842909336090088,
1620
+ "rewards/accuracies": 0.5625,
1621
+ "rewards/chosen": 0.0873967856168747,
1622
+ "rewards/margins": 0.020737329497933388,
1623
+ "rewards/rejected": 0.06665945053100586,
1624
+ "step": 980
1625
+ },
1626
+ {
1627
+ "epoch": 0.792,
1628
+ "grad_norm": 4.097748279571533,
1629
+ "learning_rate": 6.715555555555556e-07,
1630
+ "logits/chosen": 1.6006208658218384,
1631
+ "logits/rejected": 1.6803340911865234,
1632
+ "logps/chosen": -163.8089141845703,
1633
+ "logps/rejected": -167.1127166748047,
1634
+ "loss": 0.6833163261413574,
1635
+ "rewards/accuracies": 0.5874999761581421,
1636
+ "rewards/chosen": 0.07443337142467499,
1637
+ "rewards/margins": 0.023210588842630386,
1638
+ "rewards/rejected": 0.0512227788567543,
1639
+ "step": 990
1640
+ },
1641
+ {
1642
+ "epoch": 0.8,
1643
+ "grad_norm": 3.959730625152588,
1644
+ "learning_rate": 6.67111111111111e-07,
1645
+ "logits/chosen": 1.5806870460510254,
1646
+ "logits/rejected": 1.5673385858535767,
1647
+ "logps/chosen": -148.22584533691406,
1648
+ "logps/rejected": -123.79376220703125,
1649
+ "loss": 0.6892601490020752,
1650
+ "rewards/accuracies": 0.512499988079071,
1651
+ "rewards/chosen": 0.07928630709648132,
1652
+ "rewards/margins": 0.009630966000258923,
1653
+ "rewards/rejected": 0.06965534389019012,
1654
+ "step": 1000
1655
+ },
1656
+ {
1657
+ "epoch": 0.8,
1658
+ "eval_logits/chosen": 1.522994875907898,
1659
+ "eval_logits/rejected": 1.5511444807052612,
1660
+ "eval_logps/chosen": -152.6211395263672,
1661
+ "eval_logps/rejected": -147.31727600097656,
1662
+ "eval_loss": 0.6889244914054871,
1663
+ "eval_rewards/accuracies": 0.527999997138977,
1664
+ "eval_rewards/chosen": 0.07080094516277313,
1665
+ "eval_rewards/margins": 0.012245929799973965,
1666
+ "eval_rewards/rejected": 0.058555010706186295,
1667
+ "eval_runtime": 90.2821,
1668
+ "eval_samples_per_second": 5.538,
1669
+ "eval_steps_per_second": 2.769,
1670
+ "step": 1000
1671
+ },
1672
+ {
1673
+ "epoch": 0.808,
1674
+ "grad_norm": 4.409601211547852,
1675
+ "learning_rate": 6.626666666666666e-07,
1676
+ "logits/chosen": 1.6469438076019287,
1677
+ "logits/rejected": 1.6551824808120728,
1678
+ "logps/chosen": -153.6597442626953,
1679
+ "logps/rejected": -140.854248046875,
1680
+ "loss": 0.6809319496154785,
1681
+ "rewards/accuracies": 0.6499999761581421,
1682
+ "rewards/chosen": 0.09710784256458282,
1683
+ "rewards/margins": 0.026540305465459824,
1684
+ "rewards/rejected": 0.0705675408244133,
1685
+ "step": 1010
1686
+ },
1687
+ {
1688
+ "epoch": 0.816,
1689
+ "grad_norm": 6.559939384460449,
1690
+ "learning_rate": 6.582222222222222e-07,
1691
+ "logits/chosen": 1.5092687606811523,
1692
+ "logits/rejected": 1.613526701927185,
1693
+ "logps/chosen": -141.97103881835938,
1694
+ "logps/rejected": -145.5518035888672,
1695
+ "loss": 0.6865349292755127,
1696
+ "rewards/accuracies": 0.5874999761581421,
1697
+ "rewards/chosen": 0.0591324046254158,
1698
+ "rewards/margins": 0.017797131091356277,
1699
+ "rewards/rejected": 0.041335273534059525,
1700
+ "step": 1020
1701
+ },
1702
+ {
1703
+ "epoch": 0.824,
1704
+ "grad_norm": 5.1169047355651855,
1705
+ "learning_rate": 6.537777777777778e-07,
1706
+ "logits/chosen": 1.6573750972747803,
1707
+ "logits/rejected": 1.4447792768478394,
1708
+ "logps/chosen": -161.51220703125,
1709
+ "logps/rejected": -135.18309020996094,
1710
+ "loss": 0.6804659366607666,
1711
+ "rewards/accuracies": 0.6000000238418579,
1712
+ "rewards/chosen": 0.10412336885929108,
1713
+ "rewards/margins": 0.02856394089758396,
1714
+ "rewards/rejected": 0.07555942982435226,
1715
+ "step": 1030
1716
+ },
1717
+ {
1718
+ "epoch": 0.832,
1719
+ "grad_norm": 4.2392072677612305,
1720
+ "learning_rate": 6.493333333333333e-07,
1721
+ "logits/chosen": 1.6003319025039673,
1722
+ "logits/rejected": 1.6346886157989502,
1723
+ "logps/chosen": -139.08448791503906,
1724
+ "logps/rejected": -139.89825439453125,
1725
+ "loss": 0.6737568378448486,
1726
+ "rewards/accuracies": 0.5874999761581421,
1727
+ "rewards/chosen": 0.11095432937145233,
1728
+ "rewards/margins": 0.043066851794719696,
1729
+ "rewards/rejected": 0.06788748502731323,
1730
+ "step": 1040
1731
+ },
1732
+ {
1733
+ "epoch": 0.84,
1734
+ "grad_norm": 5.133569240570068,
1735
+ "learning_rate": 6.448888888888889e-07,
1736
+ "logits/chosen": 1.4209873676300049,
1737
+ "logits/rejected": 1.5513734817504883,
1738
+ "logps/chosen": -141.1917266845703,
1739
+ "logps/rejected": -130.70431518554688,
1740
+ "loss": 0.6861515998840332,
1741
+ "rewards/accuracies": 0.6000000238418579,
1742
+ "rewards/chosen": 0.10248366743326187,
1743
+ "rewards/margins": 0.016522446647286415,
1744
+ "rewards/rejected": 0.08596121519804001,
1745
+ "step": 1050
1746
+ },
1747
+ {
1748
+ "epoch": 0.848,
1749
+ "grad_norm": 4.0574445724487305,
1750
+ "learning_rate": 6.404444444444444e-07,
1751
+ "logits/chosen": 1.439145803451538,
1752
+ "logits/rejected": 1.5661519765853882,
1753
+ "logps/chosen": -111.7681884765625,
1754
+ "logps/rejected": -126.37353515625,
1755
+ "loss": 0.6691905975341796,
1756
+ "rewards/accuracies": 0.637499988079071,
1757
+ "rewards/chosen": 0.15403084456920624,
1758
+ "rewards/margins": 0.05248977616429329,
1759
+ "rewards/rejected": 0.10154107958078384,
1760
+ "step": 1060
1761
+ },
1762
+ {
1763
+ "epoch": 0.856,
1764
+ "grad_norm": 4.832082748413086,
1765
+ "learning_rate": 6.36e-07,
1766
+ "logits/chosen": 1.4902942180633545,
1767
+ "logits/rejected": 1.3948299884796143,
1768
+ "logps/chosen": -153.6760711669922,
1769
+ "logps/rejected": -147.40023803710938,
1770
+ "loss": 0.6729560375213623,
1771
+ "rewards/accuracies": 0.675000011920929,
1772
+ "rewards/chosen": 0.1259043663740158,
1773
+ "rewards/margins": 0.0469796285033226,
1774
+ "rewards/rejected": 0.0789247453212738,
1775
+ "step": 1070
1776
+ },
1777
+ {
1778
+ "epoch": 0.864,
1779
+ "grad_norm": 3.609558343887329,
1780
+ "learning_rate": 6.315555555555555e-07,
1781
+ "logits/chosen": 1.5476783514022827,
1782
+ "logits/rejected": 1.6365705728530884,
1783
+ "logps/chosen": -148.82101440429688,
1784
+ "logps/rejected": -122.3703842163086,
1785
+ "loss": 0.6743530750274658,
1786
+ "rewards/accuracies": 0.637499988079071,
1787
+ "rewards/chosen": 0.1420261561870575,
1788
+ "rewards/margins": 0.041455820202827454,
1789
+ "rewards/rejected": 0.10057034343481064,
1790
+ "step": 1080
1791
+ },
1792
+ {
1793
+ "epoch": 0.872,
1794
+ "grad_norm": 4.052758693695068,
1795
+ "learning_rate": 6.27111111111111e-07,
1796
+ "logits/chosen": 1.7934048175811768,
1797
+ "logits/rejected": 1.7036349773406982,
1798
+ "logps/chosen": -174.28765869140625,
1799
+ "logps/rejected": -183.97897338867188,
1800
+ "loss": 0.6748029708862304,
1801
+ "rewards/accuracies": 0.612500011920929,
1802
+ "rewards/chosen": 0.1439160704612732,
1803
+ "rewards/margins": 0.0408162847161293,
1804
+ "rewards/rejected": 0.1030997782945633,
1805
+ "step": 1090
1806
+ },
1807
+ {
1808
+ "epoch": 0.88,
1809
+ "grad_norm": 5.463155269622803,
1810
+ "learning_rate": 6.226666666666667e-07,
1811
+ "logits/chosen": 1.5579731464385986,
1812
+ "logits/rejected": 1.5956088304519653,
1813
+ "logps/chosen": -138.96115112304688,
1814
+ "logps/rejected": -152.51766967773438,
1815
+ "loss": 0.6884016513824462,
1816
+ "rewards/accuracies": 0.5,
1817
+ "rewards/chosen": 0.11226280778646469,
1818
+ "rewards/margins": 0.013228577561676502,
1819
+ "rewards/rejected": 0.09903421252965927,
1820
+ "step": 1100
1821
+ },
1822
+ {
1823
+ "epoch": 0.88,
1824
+ "eval_logits/chosen": 1.5295906066894531,
1825
+ "eval_logits/rejected": 1.5565518140792847,
1826
+ "eval_logps/chosen": -152.15892028808594,
1827
+ "eval_logps/rejected": -146.88125610351562,
1828
+ "eval_loss": 0.6887561678886414,
1829
+ "eval_rewards/accuracies": 0.5239999890327454,
1830
+ "eval_rewards/chosen": 0.11702151596546173,
1831
+ "eval_rewards/margins": 0.014860817231237888,
1832
+ "eval_rewards/rejected": 0.10216069966554642,
1833
+ "eval_runtime": 90.4847,
1834
+ "eval_samples_per_second": 5.526,
1835
+ "eval_steps_per_second": 2.763,
1836
+ "step": 1100
1837
+ },
1838
+ {
1839
+ "epoch": 0.888,
1840
+ "grad_norm": 5.542585372924805,
1841
+ "learning_rate": 6.182222222222222e-07,
1842
+ "logits/chosen": 1.6245231628417969,
1843
+ "logits/rejected": 1.600940465927124,
1844
+ "logps/chosen": -155.20175170898438,
1845
+ "logps/rejected": -144.58438110351562,
1846
+ "loss": 0.6818556308746337,
1847
+ "rewards/accuracies": 0.550000011920929,
1848
+ "rewards/chosen": 0.1249178797006607,
1849
+ "rewards/margins": 0.027298670262098312,
1850
+ "rewards/rejected": 0.0976191908121109,
1851
+ "step": 1110
1852
+ },
1853
+ {
1854
+ "epoch": 0.896,
1855
+ "grad_norm": 3.8583486080169678,
1856
+ "learning_rate": 6.137777777777778e-07,
1857
+ "logits/chosen": 1.6029832363128662,
1858
+ "logits/rejected": 1.652834177017212,
1859
+ "logps/chosen": -155.6839141845703,
1860
+ "logps/rejected": -149.71646118164062,
1861
+ "loss": 0.6816732883453369,
1862
+ "rewards/accuracies": 0.6499999761581421,
1863
+ "rewards/chosen": 0.13532081246376038,
1864
+ "rewards/margins": 0.025635983794927597,
1865
+ "rewards/rejected": 0.10968482494354248,
1866
+ "step": 1120
1867
+ },
1868
+ {
1869
+ "epoch": 0.904,
1870
+ "grad_norm": 4.535235404968262,
1871
+ "learning_rate": 6.093333333333332e-07,
1872
+ "logits/chosen": 1.7116715908050537,
1873
+ "logits/rejected": 1.5788238048553467,
1874
+ "logps/chosen": -146.831787109375,
1875
+ "logps/rejected": -134.20765686035156,
1876
+ "loss": 0.6897575855255127,
1877
+ "rewards/accuracies": 0.550000011920929,
1878
+ "rewards/chosen": 0.12892299890518188,
1879
+ "rewards/margins": 0.011926446110010147,
1880
+ "rewards/rejected": 0.11699654906988144,
1881
+ "step": 1130
1882
+ },
1883
+ {
1884
+ "epoch": 0.912,
1885
+ "grad_norm": 4.356500148773193,
1886
+ "learning_rate": 6.048888888888889e-07,
1887
+ "logits/chosen": 1.6915569305419922,
1888
+ "logits/rejected": 1.6864850521087646,
1889
+ "logps/chosen": -154.83705139160156,
1890
+ "logps/rejected": -141.92440795898438,
1891
+ "loss": 0.677583646774292,
1892
+ "rewards/accuracies": 0.5874999761581421,
1893
+ "rewards/chosen": 0.13437099754810333,
1894
+ "rewards/margins": 0.03629336506128311,
1895
+ "rewards/rejected": 0.09807763993740082,
1896
+ "step": 1140
1897
+ },
1898
+ {
1899
+ "epoch": 0.92,
1900
+ "grad_norm": 4.3424553871154785,
1901
+ "learning_rate": 6.004444444444444e-07,
1902
+ "logits/chosen": 1.3488795757293701,
1903
+ "logits/rejected": 1.3907456398010254,
1904
+ "logps/chosen": -138.03089904785156,
1905
+ "logps/rejected": -129.59719848632812,
1906
+ "loss": 0.6962613582611084,
1907
+ "rewards/accuracies": 0.4625000059604645,
1908
+ "rewards/chosen": 0.0907142236828804,
1909
+ "rewards/margins": 0.0030510523356497288,
1910
+ "rewards/rejected": 0.08766315877437592,
1911
+ "step": 1150
1912
+ },
1913
+ {
1914
+ "epoch": 0.928,
1915
+ "grad_norm": 4.354366779327393,
1916
+ "learning_rate": 5.96e-07,
1917
+ "logits/chosen": 1.6015634536743164,
1918
+ "logits/rejected": 1.5302627086639404,
1919
+ "logps/chosen": -125.86180114746094,
1920
+ "logps/rejected": -113.89128112792969,
1921
+ "loss": 0.6872058868408203,
1922
+ "rewards/accuracies": 0.5625,
1923
+ "rewards/chosen": 0.11558832228183746,
1924
+ "rewards/margins": 0.016637511551380157,
1925
+ "rewards/rejected": 0.09895080327987671,
1926
+ "step": 1160
1927
+ },
1928
+ {
1929
+ "epoch": 0.936,
1930
+ "grad_norm": 6.197093486785889,
1931
+ "learning_rate": 5.915555555555555e-07,
1932
+ "logits/chosen": 1.8193966150283813,
1933
+ "logits/rejected": 1.7454732656478882,
1934
+ "logps/chosen": -155.8551788330078,
1935
+ "logps/rejected": -165.95828247070312,
1936
+ "loss": 0.6903214454650879,
1937
+ "rewards/accuracies": 0.5625,
1938
+ "rewards/chosen": 0.12587358057498932,
1939
+ "rewards/margins": 0.01157000008970499,
1940
+ "rewards/rejected": 0.1143035739660263,
1941
+ "step": 1170
1942
+ },
1943
+ {
1944
+ "epoch": 0.944,
1945
+ "grad_norm": 6.9796624183654785,
1946
+ "learning_rate": 5.871111111111112e-07,
1947
+ "logits/chosen": 1.7167119979858398,
1948
+ "logits/rejected": 1.693549394607544,
1949
+ "logps/chosen": -185.20008850097656,
1950
+ "logps/rejected": -150.20620727539062,
1951
+ "loss": 0.6912973880767822,
1952
+ "rewards/accuracies": 0.512499988079071,
1953
+ "rewards/chosen": 0.1253143846988678,
1954
+ "rewards/margins": 0.009792610071599483,
1955
+ "rewards/rejected": 0.11552176624536514,
1956
+ "step": 1180
1957
+ },
1958
+ {
1959
+ "epoch": 0.952,
1960
+ "grad_norm": 4.833356857299805,
1961
+ "learning_rate": 5.826666666666666e-07,
1962
+ "logits/chosen": 1.7434288263320923,
1963
+ "logits/rejected": 1.7330595254898071,
1964
+ "logps/chosen": -165.2196044921875,
1965
+ "logps/rejected": -195.1993408203125,
1966
+ "loss": 0.6812029361724854,
1967
+ "rewards/accuracies": 0.6000000238418579,
1968
+ "rewards/chosen": 0.11465537548065186,
1969
+ "rewards/margins": 0.028904888778924942,
1970
+ "rewards/rejected": 0.08575049787759781,
1971
+ "step": 1190
1972
+ },
1973
+ {
1974
+ "epoch": 0.96,
1975
+ "grad_norm": 4.727373123168945,
1976
+ "learning_rate": 5.782222222222221e-07,
1977
+ "logits/chosen": 1.6402454376220703,
1978
+ "logits/rejected": 1.6071112155914307,
1979
+ "logps/chosen": -138.0218963623047,
1980
+ "logps/rejected": -144.56817626953125,
1981
+ "loss": 0.694274616241455,
1982
+ "rewards/accuracies": 0.550000011920929,
1983
+ "rewards/chosen": 0.11662209033966064,
1984
+ "rewards/margins": 0.003029861254617572,
1985
+ "rewards/rejected": 0.1135922223329544,
1986
+ "step": 1200
1987
+ },
1988
+ {
1989
+ "epoch": 0.96,
1990
+ "eval_logits/chosen": 1.5258080959320068,
1991
+ "eval_logits/rejected": 1.5528327226638794,
1992
+ "eval_logps/chosen": -152.21652221679688,
1993
+ "eval_logps/rejected": -146.95628356933594,
1994
+ "eval_loss": 0.6881142854690552,
1995
+ "eval_rewards/accuracies": 0.5239999890327454,
1996
+ "eval_rewards/chosen": 0.11126487702131271,
1997
+ "eval_rewards/margins": 0.01660888083279133,
1998
+ "eval_rewards/rejected": 0.09465599805116653,
1999
+ "eval_runtime": 90.2009,
2000
+ "eval_samples_per_second": 5.543,
2001
+ "eval_steps_per_second": 2.772,
2002
+ "step": 1200
2003
+ },
2004
+ {
2005
+ "epoch": 0.968,
2006
+ "grad_norm": 4.669800758361816,
2007
+ "learning_rate": 5.737777777777778e-07,
2008
+ "logits/chosen": 1.469012975692749,
2009
+ "logits/rejected": 1.4835999011993408,
2010
+ "logps/chosen": -153.94541931152344,
2011
+ "logps/rejected": -140.69659423828125,
2012
+ "loss": 0.6921857357025146,
2013
+ "rewards/accuracies": 0.4625000059604645,
2014
+ "rewards/chosen": 0.12547791004180908,
2015
+ "rewards/margins": 0.005948380567133427,
2016
+ "rewards/rejected": 0.11952953040599823,
2017
+ "step": 1210
2018
+ },
2019
+ {
2020
+ "epoch": 0.976,
2021
+ "grad_norm": 5.349202632904053,
2022
+ "learning_rate": 5.693333333333333e-07,
2023
+ "logits/chosen": 1.5056023597717285,
2024
+ "logits/rejected": 1.4966309070587158,
2025
+ "logps/chosen": -144.36300659179688,
2026
+ "logps/rejected": -122.79240417480469,
2027
+ "loss": 0.695373821258545,
2028
+ "rewards/accuracies": 0.4124999940395355,
2029
+ "rewards/chosen": 0.09502485394477844,
2030
+ "rewards/margins": -0.0005314469453878701,
2031
+ "rewards/rejected": 0.09555630385875702,
2032
+ "step": 1220
2033
+ },
2034
+ {
2035
+ "epoch": 0.984,
2036
+ "grad_norm": 5.629171848297119,
2037
+ "learning_rate": 5.648888888888889e-07,
2038
+ "logits/chosen": 1.5664876699447632,
2039
+ "logits/rejected": 1.7090803384780884,
2040
+ "logps/chosen": -136.5504150390625,
2041
+ "logps/rejected": -148.81802368164062,
2042
+ "loss": 0.6888413906097413,
2043
+ "rewards/accuracies": 0.6000000238418579,
2044
+ "rewards/chosen": 0.13565854728221893,
2045
+ "rewards/margins": 0.013341712765395641,
2046
+ "rewards/rejected": 0.12231683731079102,
2047
+ "step": 1230
2048
+ },
2049
+ {
2050
+ "epoch": 0.992,
2051
+ "grad_norm": 5.568618297576904,
2052
+ "learning_rate": 5.604444444444444e-07,
2053
+ "logits/chosen": 1.512286901473999,
2054
+ "logits/rejected": 1.7082159519195557,
2055
+ "logps/chosen": -143.9463653564453,
2056
+ "logps/rejected": -146.82052612304688,
2057
+ "loss": 0.6722752571105957,
2058
+ "rewards/accuracies": 0.5874999761581421,
2059
+ "rewards/chosen": 0.12829402089118958,
2060
+ "rewards/margins": 0.046781741082668304,
2061
+ "rewards/rejected": 0.08151227235794067,
2062
+ "step": 1240
2063
+ },
2064
+ {
2065
+ "epoch": 1.0,
2066
+ "grad_norm": 4.786505699157715,
2067
+ "learning_rate": 5.560000000000001e-07,
2068
+ "logits/chosen": 1.6615822315216064,
2069
+ "logits/rejected": 1.7368179559707642,
2070
+ "logps/chosen": -144.40292358398438,
2071
+ "logps/rejected": -157.58763122558594,
2072
+ "loss": 0.6829993724822998,
2073
+ "rewards/accuracies": 0.5625,
2074
+ "rewards/chosen": 0.11200229823589325,
2075
+ "rewards/margins": 0.026959875598549843,
2076
+ "rewards/rejected": 0.08504240959882736,
2077
+ "step": 1250
2078
+ },
2079
+ {
2080
+ "epoch": 1.008,
2081
+ "grad_norm": 4.871355056762695,
2082
+ "learning_rate": 5.515555555555555e-07,
2083
+ "logits/chosen": 1.581756830215454,
2084
+ "logits/rejected": 1.643133521080017,
2085
+ "logps/chosen": -123.44065856933594,
2086
+ "logps/rejected": -131.16763305664062,
2087
+ "loss": 0.6869094848632813,
2088
+ "rewards/accuracies": 0.48750001192092896,
2089
+ "rewards/chosen": 0.11280514299869537,
2090
+ "rewards/margins": 0.016064399853348732,
2091
+ "rewards/rejected": 0.09674074500799179,
2092
+ "step": 1260
2093
+ },
2094
+ {
2095
+ "epoch": 1.016,
2096
+ "grad_norm": 5.43352746963501,
2097
+ "learning_rate": 5.471111111111111e-07,
2098
+ "logits/chosen": 1.8056955337524414,
2099
+ "logits/rejected": 1.7294971942901611,
2100
+ "logps/chosen": -159.89236450195312,
2101
+ "logps/rejected": -152.61293029785156,
2102
+ "loss": 0.6701028347015381,
2103
+ "rewards/accuracies": 0.625,
2104
+ "rewards/chosen": 0.1585489809513092,
2105
+ "rewards/margins": 0.0533718541264534,
2106
+ "rewards/rejected": 0.1051771491765976,
2107
+ "step": 1270
2108
+ },
2109
+ {
2110
+ "epoch": 1.024,
2111
+ "grad_norm": 5.0044474601745605,
2112
+ "learning_rate": 5.426666666666666e-07,
2113
+ "logits/chosen": 1.68048894405365,
2114
+ "logits/rejected": 1.6845731735229492,
2115
+ "logps/chosen": -153.21994018554688,
2116
+ "logps/rejected": -140.06024169921875,
2117
+ "loss": 0.675870132446289,
2118
+ "rewards/accuracies": 0.5874999761581421,
2119
+ "rewards/chosen": 0.11108388751745224,
2120
+ "rewards/margins": 0.039517562836408615,
2121
+ "rewards/rejected": 0.07156632840633392,
2122
+ "step": 1280
2123
+ },
2124
+ {
2125
+ "epoch": 1.032,
2126
+ "grad_norm": 4.051770210266113,
2127
+ "learning_rate": 5.382222222222223e-07,
2128
+ "logits/chosen": 1.6341949701309204,
2129
+ "logits/rejected": 1.7052574157714844,
2130
+ "logps/chosen": -143.89512634277344,
2131
+ "logps/rejected": -132.5374298095703,
2132
+ "loss": 0.6808773040771484,
2133
+ "rewards/accuracies": 0.5625,
2134
+ "rewards/chosen": 0.11111323535442352,
2135
+ "rewards/margins": 0.02908928692340851,
2136
+ "rewards/rejected": 0.08202396333217621,
2137
+ "step": 1290
2138
+ },
2139
+ {
2140
+ "epoch": 1.04,
2141
+ "grad_norm": 4.0852885246276855,
2142
+ "learning_rate": 5.337777777777778e-07,
2143
+ "logits/chosen": 1.701703429222107,
2144
+ "logits/rejected": 1.5303716659545898,
2145
+ "logps/chosen": -149.3526611328125,
2146
+ "logps/rejected": -129.2085418701172,
2147
+ "loss": 0.6718564987182617,
2148
+ "rewards/accuracies": 0.550000011920929,
2149
+ "rewards/chosen": 0.13976502418518066,
2150
+ "rewards/margins": 0.04783231392502785,
2151
+ "rewards/rejected": 0.09193271398544312,
2152
+ "step": 1300
2153
+ },
2154
+ {
2155
+ "epoch": 1.04,
2156
+ "eval_logits/chosen": 1.5326013565063477,
2157
+ "eval_logits/rejected": 1.5598564147949219,
2158
+ "eval_logps/chosen": -152.15328979492188,
2159
+ "eval_logps/rejected": -146.90542602539062,
2160
+ "eval_loss": 0.6874103546142578,
2161
+ "eval_rewards/accuracies": 0.5460000038146973,
2162
+ "eval_rewards/chosen": 0.1175844818353653,
2163
+ "eval_rewards/margins": 0.01784202829003334,
2164
+ "eval_rewards/rejected": 0.09974244982004166,
2165
+ "eval_runtime": 90.5137,
2166
+ "eval_samples_per_second": 5.524,
2167
+ "eval_steps_per_second": 2.762,
2168
+ "step": 1300
2169
+ }
2170
+ ],
2171
+ "logging_steps": 10,
2172
+ "max_steps": 2500,
2173
+ "num_input_tokens_seen": 0,
2174
+ "num_train_epochs": 2,
2175
+ "save_steps": 100,
2176
+ "stateful_callbacks": {
2177
+ "TrainerControl": {
2178
+ "args": {
2179
+ "should_epoch_stop": false,
2180
+ "should_evaluate": false,
2181
+ "should_log": false,
2182
+ "should_save": true,
2183
+ "should_training_stop": false
2184
+ },
2185
+ "attributes": {}
2186
+ }
2187
+ },
2188
+ "total_flos": 0.0,
2189
+ "train_batch_size": 2,
2190
+ "trial_name": null,
2191
+ "trial_params": null
2192
+ }
v5/DPO/DPO_10k/lora/checkpoint-1300/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:677b288b67816c6ab7a9dcdd40d26bcb142fa3ad3ad050eaeeb4b73a1ba4b498
3
+ size 6161
v5/DPO/DPO_10k/lora/checkpoint-2400/README.md ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: meta-llama/Llama-3.2-1B-Instruct
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:meta-llama/Llama-3.2-1B-Instruct
7
+ - dpo
8
+ - lora
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
v5/DPO/DPO_10k/lora/checkpoint-2400/adapter_config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.2-1B-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.1,
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": 64,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "o_proj",
33
+ "q_proj",
34
+ "down_proj",
35
+ "k_proj",
36
+ "gate_proj",
37
+ "up_proj",
38
+ "v_proj"
39
+ ],
40
+ "target_parameters": null,
41
+ "task_type": "CAUSAL_LM",
42
+ "trainable_token_indices": null,
43
+ "use_dora": false,
44
+ "use_qalora": false,
45
+ "use_rslora": false
46
+ }
v5/DPO/DPO_10k/lora/checkpoint-2400/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:925fb02b11fdf9941bdfae0889066e9dab940f09395d0149a16abe025b37ad15
3
+ size 180385008
v5/DPO/DPO_10k/lora/checkpoint-2400/chat_template.jinja ADDED
@@ -0,0 +1,93 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ {%- if strftime_now is defined %}
10
+ {%- set date_string = strftime_now("%d %b %Y") %}
11
+ {%- else %}
12
+ {%- set date_string = "26 Jul 2024" %}
13
+ {%- endif %}
14
+ {%- endif %}
15
+ {%- if not tools is defined %}
16
+ {%- set tools = none %}
17
+ {%- endif %}
18
+
19
+ {#- This block extracts the system message, so we can slot it into the right place. #}
20
+ {%- if messages[0]['role'] == 'system' %}
21
+ {%- set system_message = messages[0]['content']|trim %}
22
+ {%- set messages = messages[1:] %}
23
+ {%- else %}
24
+ {%- set system_message = "" %}
25
+ {%- endif %}
26
+
27
+ {#- System message #}
28
+ {{- "<|start_header_id|>system<|end_header_id|>\n\n" }}
29
+ {%- if tools is not none %}
30
+ {{- "Environment: ipython\n" }}
31
+ {%- endif %}
32
+ {{- "Cutting Knowledge Date: December 2023\n" }}
33
+ {{- "Today Date: " + date_string + "\n\n" }}
34
+ {%- if tools is not none and not tools_in_user_message %}
35
+ {{- "You have access to the following functions. To call a function, please respond with JSON for a function call." }}
36
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
37
+ {{- "Do not use variables.\n\n" }}
38
+ {%- for t in tools %}
39
+ {{- t | tojson(indent=4) }}
40
+ {{- "\n\n" }}
41
+ {%- endfor %}
42
+ {%- endif %}
43
+ {{- system_message }}
44
+ {{- "<|eot_id|>" }}
45
+
46
+ {#- Custom tools are passed in a user message with some extra guidance #}
47
+ {%- if tools_in_user_message and not tools is none %}
48
+ {#- Extract the first user message so we can plug it in here #}
49
+ {%- if messages | length != 0 %}
50
+ {%- set first_user_message = messages[0]['content']|trim %}
51
+ {%- set messages = messages[1:] %}
52
+ {%- else %}
53
+ {{- raise_exception("Cannot put tools in the first user message when there's no first user message!") }}
54
+ {%- endif %}
55
+ {{- '<|start_header_id|>user<|end_header_id|>\n\n' -}}
56
+ {{- "Given the following functions, please respond with a JSON for a function call " }}
57
+ {{- "with its proper arguments that best answers the given prompt.\n\n" }}
58
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
59
+ {{- "Do not use variables.\n\n" }}
60
+ {%- for t in tools %}
61
+ {{- t | tojson(indent=4) }}
62
+ {{- "\n\n" }}
63
+ {%- endfor %}
64
+ {{- first_user_message + "<|eot_id|>"}}
65
+ {%- endif %}
66
+
67
+ {%- for message in messages %}
68
+ {%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}
69
+ {{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'+ message['content'] | trim + '<|eot_id|>' }}
70
+ {%- elif 'tool_calls' in message %}
71
+ {%- if not message.tool_calls|length == 1 %}
72
+ {{- raise_exception("This model only supports single tool-calls at once!") }}
73
+ {%- endif %}
74
+ {%- set tool_call = message.tool_calls[0].function %}
75
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' -}}
76
+ {{- '{"name": "' + tool_call.name + '", ' }}
77
+ {{- '"parameters": ' }}
78
+ {{- tool_call.arguments | tojson }}
79
+ {{- "}" }}
80
+ {{- "<|eot_id|>" }}
81
+ {%- elif message.role == "tool" or message.role == "ipython" %}
82
+ {{- "<|start_header_id|>ipython<|end_header_id|>\n\n" }}
83
+ {%- if message.content is mapping or message.content is iterable %}
84
+ {{- message.content | tojson }}
85
+ {%- else %}
86
+ {{- message.content }}
87
+ {%- endif %}
88
+ {{- "<|eot_id|>" }}
89
+ {%- endif %}
90
+ {%- endfor %}
91
+ {%- if add_generation_prompt %}
92
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' }}
93
+ {%- endif %}
v5/DPO/DPO_10k/lora/checkpoint-2400/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:98ac224d5ed27ace528190a6990080c745116a4b920d737d850dcbf7cec83638
3
+ size 360902475
v5/DPO/DPO_10k/lora/checkpoint-2400/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7affab63b271ed0f59a5b53056fc0a581226a41dcdf2fc2b80b669e7c3cf714
3
+ size 14645
v5/DPO/DPO_10k/lora/checkpoint-2400/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ee3be4931d9e7c32cc76f610ed77ea9a83d47b5639a53431c51dcc937f80ff5
3
+ size 1383
v5/DPO/DPO_10k/lora/checkpoint-2400/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:050d6987b00cc915a861681f6abd4fe1d6f88b067e04461ce21e4f85653a111f
3
+ size 1465
v5/DPO/DPO_10k/lora/checkpoint-2400/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b9e4e7fb171f92fd137b777cc2714bf87d11576700a1dcd7a399e7bbe39537b
3
+ size 17209920
v5/DPO/DPO_10k/lora/checkpoint-2400/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
+ }
v5/DPO/DPO_10k/lora/checkpoint-2400/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
v5/DPO/DPO_10k/lora/checkpoint-2400/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:677b288b67816c6ab7a9dcdd40d26bcb142fa3ad3ad050eaeeb4b73a1ba4b498
3
+ size 6161
v5/DPO/DPO_10k/lora/checkpoint-2500/README.md ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: meta-llama/Llama-3.2-1B-Instruct
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:meta-llama/Llama-3.2-1B-Instruct
7
+ - dpo
8
+ - lora
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
v5/DPO/DPO_10k/lora/checkpoint-2500/adapter_config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.2-1B-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.1,
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": 64,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "o_proj",
33
+ "q_proj",
34
+ "down_proj",
35
+ "k_proj",
36
+ "gate_proj",
37
+ "up_proj",
38
+ "v_proj"
39
+ ],
40
+ "target_parameters": null,
41
+ "task_type": "CAUSAL_LM",
42
+ "trainable_token_indices": null,
43
+ "use_dora": false,
44
+ "use_qalora": false,
45
+ "use_rslora": false
46
+ }
v5/DPO/DPO_10k/lora/checkpoint-2500/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a0bbd3c3e4c00bee837cf13980be86026bc37eed7182fa43a7b6a34481dc706f
3
+ size 180385008
v5/DPO/DPO_10k/lora/checkpoint-2500/chat_template.jinja ADDED
@@ -0,0 +1,93 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ {%- if strftime_now is defined %}
10
+ {%- set date_string = strftime_now("%d %b %Y") %}
11
+ {%- else %}
12
+ {%- set date_string = "26 Jul 2024" %}
13
+ {%- endif %}
14
+ {%- endif %}
15
+ {%- if not tools is defined %}
16
+ {%- set tools = none %}
17
+ {%- endif %}
18
+
19
+ {#- This block extracts the system message, so we can slot it into the right place. #}
20
+ {%- if messages[0]['role'] == 'system' %}
21
+ {%- set system_message = messages[0]['content']|trim %}
22
+ {%- set messages = messages[1:] %}
23
+ {%- else %}
24
+ {%- set system_message = "" %}
25
+ {%- endif %}
26
+
27
+ {#- System message #}
28
+ {{- "<|start_header_id|>system<|end_header_id|>\n\n" }}
29
+ {%- if tools is not none %}
30
+ {{- "Environment: ipython\n" }}
31
+ {%- endif %}
32
+ {{- "Cutting Knowledge Date: December 2023\n" }}
33
+ {{- "Today Date: " + date_string + "\n\n" }}
34
+ {%- if tools is not none and not tools_in_user_message %}
35
+ {{- "You have access to the following functions. To call a function, please respond with JSON for a function call." }}
36
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
37
+ {{- "Do not use variables.\n\n" }}
38
+ {%- for t in tools %}
39
+ {{- t | tojson(indent=4) }}
40
+ {{- "\n\n" }}
41
+ {%- endfor %}
42
+ {%- endif %}
43
+ {{- system_message }}
44
+ {{- "<|eot_id|>" }}
45
+
46
+ {#- Custom tools are passed in a user message with some extra guidance #}
47
+ {%- if tools_in_user_message and not tools is none %}
48
+ {#- Extract the first user message so we can plug it in here #}
49
+ {%- if messages | length != 0 %}
50
+ {%- set first_user_message = messages[0]['content']|trim %}
51
+ {%- set messages = messages[1:] %}
52
+ {%- else %}
53
+ {{- raise_exception("Cannot put tools in the first user message when there's no first user message!") }}
54
+ {%- endif %}
55
+ {{- '<|start_header_id|>user<|end_header_id|>\n\n' -}}
56
+ {{- "Given the following functions, please respond with a JSON for a function call " }}
57
+ {{- "with its proper arguments that best answers the given prompt.\n\n" }}
58
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
59
+ {{- "Do not use variables.\n\n" }}
60
+ {%- for t in tools %}
61
+ {{- t | tojson(indent=4) }}
62
+ {{- "\n\n" }}
63
+ {%- endfor %}
64
+ {{- first_user_message + "<|eot_id|>"}}
65
+ {%- endif %}
66
+
67
+ {%- for message in messages %}
68
+ {%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}
69
+ {{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'+ message['content'] | trim + '<|eot_id|>' }}
70
+ {%- elif 'tool_calls' in message %}
71
+ {%- if not message.tool_calls|length == 1 %}
72
+ {{- raise_exception("This model only supports single tool-calls at once!") }}
73
+ {%- endif %}
74
+ {%- set tool_call = message.tool_calls[0].function %}
75
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' -}}
76
+ {{- '{"name": "' + tool_call.name + '", ' }}
77
+ {{- '"parameters": ' }}
78
+ {{- tool_call.arguments | tojson }}
79
+ {{- "}" }}
80
+ {{- "<|eot_id|>" }}
81
+ {%- elif message.role == "tool" or message.role == "ipython" %}
82
+ {{- "<|start_header_id|>ipython<|end_header_id|>\n\n" }}
83
+ {%- if message.content is mapping or message.content is iterable %}
84
+ {{- message.content | tojson }}
85
+ {%- else %}
86
+ {{- message.content }}
87
+ {%- endif %}
88
+ {{- "<|eot_id|>" }}
89
+ {%- endif %}
90
+ {%- endfor %}
91
+ {%- if add_generation_prompt %}
92
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' }}
93
+ {%- endif %}
v5/DPO/DPO_10k/lora/checkpoint-2500/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a85a16bd9a9e3b84f02ff23b0aae6c239155a06a501d1d776ea15eba57fc952a
3
+ size 360902475
v5/DPO/DPO_10k/lora/checkpoint-2500/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d1786ad2057a678cc204dadc7fc5d1a4f939be477df219f770c7d40e9270281
3
+ size 14645
v5/DPO/DPO_10k/lora/checkpoint-2500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a4f7df58dbec84710c280fe8ef7660bf055a3d051df5a58eccd60f22b169583
3
+ size 1383
v5/DPO/DPO_10k/lora/checkpoint-2500/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7154cc492a37bc9bb0e6d09325e786c916142f1f119ecd183a60eecde4f9f283
3
+ size 1465
v5/DPO/DPO_10k/lora/checkpoint-2500/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b9e4e7fb171f92fd137b777cc2714bf87d11576700a1dcd7a399e7bbe39537b
3
+ size 17209920
v5/DPO/DPO_10k/lora/checkpoint-2500/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
+ }
v5/DPO/DPO_10k/lora/checkpoint-2500/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
v5/DPO/DPO_10k/lora/checkpoint-2500/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:677b288b67816c6ab7a9dcdd40d26bcb142fa3ad3ad050eaeeb4b73a1ba4b498
3
+ size 6161
v5/DPO/DPO_1k/DPO_1k/README.md ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: meta-llama/Llama-3.2-1B-Instruct
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:meta-llama/Llama-3.2-1B-Instruct
7
+ - dpo
8
+ - lora
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
v5/DPO/DPO_1k/DPO_1k/adapter_config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.2-1B-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.1,
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": 64,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "k_proj",
33
+ "up_proj",
34
+ "q_proj",
35
+ "down_proj",
36
+ "o_proj",
37
+ "gate_proj",
38
+ "v_proj"
39
+ ],
40
+ "target_parameters": null,
41
+ "task_type": "CAUSAL_LM",
42
+ "trainable_token_indices": null,
43
+ "use_dora": false,
44
+ "use_qalora": false,
45
+ "use_rslora": false
46
+ }
v5/DPO/DPO_1k/DPO_1k/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd39ce893511c42bb66fcc45902600954d24ae57a8646fa31ae8e964732c757a
3
+ size 180385008