Vasudevakrishna commited on
Commit
35e6e46
1 Parent(s): a8709ad

Upload 12 files

Browse files
ckpts/adapter_config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "microsoft/phi-2",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layers_pattern": null,
10
+ "layers_to_transform": null,
11
+ "loftq_config": {},
12
+ "lora_alpha": 16,
13
+ "lora_dropout": 0.1,
14
+ "megatron_config": null,
15
+ "megatron_core": "megatron.core",
16
+ "modules_to_save": null,
17
+ "peft_type": "LORA",
18
+ "r": 64,
19
+ "rank_pattern": {},
20
+ "revision": null,
21
+ "target_modules": [
22
+ "v_proj",
23
+ "fc1",
24
+ "k_proj",
25
+ "q_proj",
26
+ "dense",
27
+ "fc2"
28
+ ],
29
+ "task_type": "CAUSAL_LM",
30
+ "use_rslora": false
31
+ }
ckpts/added_tokens.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "\t\t": 50294,
3
+ "\t\t\t": 50293,
4
+ "\t\t\t\t": 50292,
5
+ "\t\t\t\t\t": 50291,
6
+ "\t\t\t\t\t\t": 50290,
7
+ "\t\t\t\t\t\t\t": 50289,
8
+ "\t\t\t\t\t\t\t\t": 50288,
9
+ "\t\t\t\t\t\t\t\t\t": 50287,
10
+ " ": 50286,
11
+ " ": 50285,
12
+ " ": 50284,
13
+ " ": 50283,
14
+ " ": 50282,
15
+ " ": 50281,
16
+ " ": 50280,
17
+ " ": 50279,
18
+ " ": 50278,
19
+ " ": 50277,
20
+ " ": 50276,
21
+ " ": 50275,
22
+ " ": 50274,
23
+ " ": 50273,
24
+ " ": 50272,
25
+ " ": 50271,
26
+ " ": 50270,
27
+ " ": 50269,
28
+ " ": 50268,
29
+ " ": 50267,
30
+ " ": 50266,
31
+ " ": 50265,
32
+ " ": 50264,
33
+ " ": 50263,
34
+ " ": 50262,
35
+ " ": 50261,
36
+ " ": 50260,
37
+ " ": 50259,
38
+ " ": 50258,
39
+ " ": 50257
40
+ }
ckpts/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
ckpts/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73e2bffa05c4cb4baf7fd6f4d9cfbd562add37f7c6e087471979c9283495ecc4
3
+ size 14244
ckpts/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb54cffa206db0c12f0795a9e4804d798628be7cb6d50b66106d7441bc5abccf
3
+ size 1064
ckpts/special_tokens_map.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<|endoftext|>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|endoftext|>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": "<|endoftext|>",
17
+ "unk_token": {
18
+ "content": "<|endoftext|>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ }
24
+ }
ckpts/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
ckpts/tokenizer_config.json ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "50256": {
5
+ "content": "<|endoftext|>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "50257": {
13
+ "content": " ",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": false
19
+ },
20
+ "50258": {
21
+ "content": " ",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": false
27
+ },
28
+ "50259": {
29
+ "content": " ",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": false
35
+ },
36
+ "50260": {
37
+ "content": " ",
38
+ "lstrip": false,
39
+ "normalized": true,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": false
43
+ },
44
+ "50261": {
45
+ "content": " ",
46
+ "lstrip": false,
47
+ "normalized": true,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": false
51
+ },
52
+ "50262": {
53
+ "content": " ",
54
+ "lstrip": false,
55
+ "normalized": true,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": false
59
+ },
60
+ "50263": {
61
+ "content": " ",
62
+ "lstrip": false,
63
+ "normalized": true,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": false
67
+ },
68
+ "50264": {
69
+ "content": " ",
70
+ "lstrip": false,
71
+ "normalized": true,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": false
75
+ },
76
+ "50265": {
77
+ "content": " ",
78
+ "lstrip": false,
79
+ "normalized": true,
80
+ "rstrip": false,
81
+ "single_word": false,
82
+ "special": false
83
+ },
84
+ "50266": {
85
+ "content": " ",
86
+ "lstrip": false,
87
+ "normalized": true,
88
+ "rstrip": false,
89
+ "single_word": false,
90
+ "special": false
91
+ },
92
+ "50267": {
93
+ "content": " ",
94
+ "lstrip": false,
95
+ "normalized": true,
96
+ "rstrip": false,
97
+ "single_word": false,
98
+ "special": false
99
+ },
100
+ "50268": {
101
+ "content": " ",
102
+ "lstrip": false,
103
+ "normalized": true,
104
+ "rstrip": false,
105
+ "single_word": false,
106
+ "special": false
107
+ },
108
+ "50269": {
109
+ "content": " ",
110
+ "lstrip": false,
111
+ "normalized": true,
112
+ "rstrip": false,
113
+ "single_word": false,
114
+ "special": false
115
+ },
116
+ "50270": {
117
+ "content": " ",
118
+ "lstrip": false,
119
+ "normalized": true,
120
+ "rstrip": false,
121
+ "single_word": false,
122
+ "special": false
123
+ },
124
+ "50271": {
125
+ "content": " ",
126
+ "lstrip": false,
127
+ "normalized": true,
128
+ "rstrip": false,
129
+ "single_word": false,
130
+ "special": false
131
+ },
132
+ "50272": {
133
+ "content": " ",
134
+ "lstrip": false,
135
+ "normalized": true,
136
+ "rstrip": false,
137
+ "single_word": false,
138
+ "special": false
139
+ },
140
+ "50273": {
141
+ "content": " ",
142
+ "lstrip": false,
143
+ "normalized": true,
144
+ "rstrip": false,
145
+ "single_word": false,
146
+ "special": false
147
+ },
148
+ "50274": {
149
+ "content": " ",
150
+ "lstrip": false,
151
+ "normalized": true,
152
+ "rstrip": false,
153
+ "single_word": false,
154
+ "special": false
155
+ },
156
+ "50275": {
157
+ "content": " ",
158
+ "lstrip": false,
159
+ "normalized": true,
160
+ "rstrip": false,
161
+ "single_word": false,
162
+ "special": false
163
+ },
164
+ "50276": {
165
+ "content": " ",
166
+ "lstrip": false,
167
+ "normalized": true,
168
+ "rstrip": false,
169
+ "single_word": false,
170
+ "special": false
171
+ },
172
+ "50277": {
173
+ "content": " ",
174
+ "lstrip": false,
175
+ "normalized": true,
176
+ "rstrip": false,
177
+ "single_word": false,
178
+ "special": false
179
+ },
180
+ "50278": {
181
+ "content": " ",
182
+ "lstrip": false,
183
+ "normalized": true,
184
+ "rstrip": false,
185
+ "single_word": false,
186
+ "special": false
187
+ },
188
+ "50279": {
189
+ "content": " ",
190
+ "lstrip": false,
191
+ "normalized": true,
192
+ "rstrip": false,
193
+ "single_word": false,
194
+ "special": false
195
+ },
196
+ "50280": {
197
+ "content": " ",
198
+ "lstrip": false,
199
+ "normalized": true,
200
+ "rstrip": false,
201
+ "single_word": false,
202
+ "special": false
203
+ },
204
+ "50281": {
205
+ "content": " ",
206
+ "lstrip": false,
207
+ "normalized": true,
208
+ "rstrip": false,
209
+ "single_word": false,
210
+ "special": false
211
+ },
212
+ "50282": {
213
+ "content": " ",
214
+ "lstrip": false,
215
+ "normalized": true,
216
+ "rstrip": false,
217
+ "single_word": false,
218
+ "special": false
219
+ },
220
+ "50283": {
221
+ "content": " ",
222
+ "lstrip": false,
223
+ "normalized": true,
224
+ "rstrip": false,
225
+ "single_word": false,
226
+ "special": false
227
+ },
228
+ "50284": {
229
+ "content": " ",
230
+ "lstrip": false,
231
+ "normalized": true,
232
+ "rstrip": false,
233
+ "single_word": false,
234
+ "special": false
235
+ },
236
+ "50285": {
237
+ "content": " ",
238
+ "lstrip": false,
239
+ "normalized": true,
240
+ "rstrip": false,
241
+ "single_word": false,
242
+ "special": false
243
+ },
244
+ "50286": {
245
+ "content": " ",
246
+ "lstrip": false,
247
+ "normalized": true,
248
+ "rstrip": false,
249
+ "single_word": false,
250
+ "special": false
251
+ },
252
+ "50287": {
253
+ "content": "\t\t\t\t\t\t\t\t\t",
254
+ "lstrip": false,
255
+ "normalized": true,
256
+ "rstrip": false,
257
+ "single_word": false,
258
+ "special": false
259
+ },
260
+ "50288": {
261
+ "content": "\t\t\t\t\t\t\t\t",
262
+ "lstrip": false,
263
+ "normalized": true,
264
+ "rstrip": false,
265
+ "single_word": false,
266
+ "special": false
267
+ },
268
+ "50289": {
269
+ "content": "\t\t\t\t\t\t\t",
270
+ "lstrip": false,
271
+ "normalized": true,
272
+ "rstrip": false,
273
+ "single_word": false,
274
+ "special": false
275
+ },
276
+ "50290": {
277
+ "content": "\t\t\t\t\t\t",
278
+ "lstrip": false,
279
+ "normalized": true,
280
+ "rstrip": false,
281
+ "single_word": false,
282
+ "special": false
283
+ },
284
+ "50291": {
285
+ "content": "\t\t\t\t\t",
286
+ "lstrip": false,
287
+ "normalized": true,
288
+ "rstrip": false,
289
+ "single_word": false,
290
+ "special": false
291
+ },
292
+ "50292": {
293
+ "content": "\t\t\t\t",
294
+ "lstrip": false,
295
+ "normalized": true,
296
+ "rstrip": false,
297
+ "single_word": false,
298
+ "special": false
299
+ },
300
+ "50293": {
301
+ "content": "\t\t\t",
302
+ "lstrip": false,
303
+ "normalized": true,
304
+ "rstrip": false,
305
+ "single_word": false,
306
+ "special": false
307
+ },
308
+ "50294": {
309
+ "content": "\t\t",
310
+ "lstrip": false,
311
+ "normalized": true,
312
+ "rstrip": false,
313
+ "single_word": false,
314
+ "special": false
315
+ }
316
+ },
317
+ "bos_token": "<|endoftext|>",
318
+ "clean_up_tokenization_spaces": true,
319
+ "eos_token": "<|endoftext|>",
320
+ "model_max_length": 2048,
321
+ "pad_token": "<|endoftext|>",
322
+ "tokenizer_class": "CodeGenTokenizer",
323
+ "unk_token": "<|endoftext|>"
324
+ }
ckpts/trainer_state.json ADDED
@@ -0,0 +1,321 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 1.0183299389002036,
5
+ "eval_steps": 500,
6
+ "global_step": 500,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.02,
13
+ "learning_rate": 0.0002,
14
+ "loss": 1.6531,
15
+ "step": 10
16
+ },
17
+ {
18
+ "epoch": 0.04,
19
+ "learning_rate": 0.0002,
20
+ "loss": 1.4496,
21
+ "step": 20
22
+ },
23
+ {
24
+ "epoch": 0.06,
25
+ "learning_rate": 0.0002,
26
+ "loss": 1.5821,
27
+ "step": 30
28
+ },
29
+ {
30
+ "epoch": 0.08,
31
+ "learning_rate": 0.0002,
32
+ "loss": 1.895,
33
+ "step": 40
34
+ },
35
+ {
36
+ "epoch": 0.1,
37
+ "learning_rate": 0.0002,
38
+ "loss": 1.9526,
39
+ "step": 50
40
+ },
41
+ {
42
+ "epoch": 0.12,
43
+ "learning_rate": 0.0002,
44
+ "loss": 1.4505,
45
+ "step": 60
46
+ },
47
+ {
48
+ "epoch": 0.14,
49
+ "learning_rate": 0.0002,
50
+ "loss": 1.4075,
51
+ "step": 70
52
+ },
53
+ {
54
+ "epoch": 0.16,
55
+ "learning_rate": 0.0002,
56
+ "loss": 1.472,
57
+ "step": 80
58
+ },
59
+ {
60
+ "epoch": 0.18,
61
+ "learning_rate": 0.0002,
62
+ "loss": 1.8758,
63
+ "step": 90
64
+ },
65
+ {
66
+ "epoch": 0.2,
67
+ "learning_rate": 0.0002,
68
+ "loss": 1.9509,
69
+ "step": 100
70
+ },
71
+ {
72
+ "epoch": 0.22,
73
+ "learning_rate": 0.0002,
74
+ "loss": 1.4616,
75
+ "step": 110
76
+ },
77
+ {
78
+ "epoch": 0.24,
79
+ "learning_rate": 0.0002,
80
+ "loss": 1.4365,
81
+ "step": 120
82
+ },
83
+ {
84
+ "epoch": 0.26,
85
+ "learning_rate": 0.0002,
86
+ "loss": 1.5214,
87
+ "step": 130
88
+ },
89
+ {
90
+ "epoch": 0.29,
91
+ "learning_rate": 0.0002,
92
+ "loss": 1.8216,
93
+ "step": 140
94
+ },
95
+ {
96
+ "epoch": 0.31,
97
+ "learning_rate": 0.0002,
98
+ "loss": 1.9155,
99
+ "step": 150
100
+ },
101
+ {
102
+ "epoch": 0.33,
103
+ "learning_rate": 0.0002,
104
+ "loss": 1.4898,
105
+ "step": 160
106
+ },
107
+ {
108
+ "epoch": 0.35,
109
+ "learning_rate": 0.0002,
110
+ "loss": 1.3915,
111
+ "step": 170
112
+ },
113
+ {
114
+ "epoch": 0.37,
115
+ "learning_rate": 0.0002,
116
+ "loss": 1.5799,
117
+ "step": 180
118
+ },
119
+ {
120
+ "epoch": 0.39,
121
+ "learning_rate": 0.0002,
122
+ "loss": 2.0476,
123
+ "step": 190
124
+ },
125
+ {
126
+ "epoch": 0.41,
127
+ "learning_rate": 0.0002,
128
+ "loss": 1.9179,
129
+ "step": 200
130
+ },
131
+ {
132
+ "epoch": 0.43,
133
+ "learning_rate": 0.0002,
134
+ "loss": 1.4597,
135
+ "step": 210
136
+ },
137
+ {
138
+ "epoch": 0.45,
139
+ "learning_rate": 0.0002,
140
+ "loss": 1.4114,
141
+ "step": 220
142
+ },
143
+ {
144
+ "epoch": 0.47,
145
+ "learning_rate": 0.0002,
146
+ "loss": 1.5009,
147
+ "step": 230
148
+ },
149
+ {
150
+ "epoch": 0.49,
151
+ "learning_rate": 0.0002,
152
+ "loss": 1.8596,
153
+ "step": 240
154
+ },
155
+ {
156
+ "epoch": 0.51,
157
+ "learning_rate": 0.0002,
158
+ "loss": 1.851,
159
+ "step": 250
160
+ },
161
+ {
162
+ "epoch": 0.53,
163
+ "learning_rate": 0.0002,
164
+ "loss": 1.3758,
165
+ "step": 260
166
+ },
167
+ {
168
+ "epoch": 0.55,
169
+ "learning_rate": 0.0002,
170
+ "loss": 1.3879,
171
+ "step": 270
172
+ },
173
+ {
174
+ "epoch": 0.57,
175
+ "learning_rate": 0.0002,
176
+ "loss": 1.6088,
177
+ "step": 280
178
+ },
179
+ {
180
+ "epoch": 0.59,
181
+ "learning_rate": 0.0002,
182
+ "loss": 1.8558,
183
+ "step": 290
184
+ },
185
+ {
186
+ "epoch": 0.61,
187
+ "learning_rate": 0.0002,
188
+ "loss": 1.8685,
189
+ "step": 300
190
+ },
191
+ {
192
+ "epoch": 0.63,
193
+ "learning_rate": 0.0002,
194
+ "loss": 1.3584,
195
+ "step": 310
196
+ },
197
+ {
198
+ "epoch": 0.65,
199
+ "learning_rate": 0.0002,
200
+ "loss": 1.3976,
201
+ "step": 320
202
+ },
203
+ {
204
+ "epoch": 0.67,
205
+ "learning_rate": 0.0002,
206
+ "loss": 1.5402,
207
+ "step": 330
208
+ },
209
+ {
210
+ "epoch": 0.69,
211
+ "learning_rate": 0.0002,
212
+ "loss": 1.8134,
213
+ "step": 340
214
+ },
215
+ {
216
+ "epoch": 0.71,
217
+ "learning_rate": 0.0002,
218
+ "loss": 1.9623,
219
+ "step": 350
220
+ },
221
+ {
222
+ "epoch": 0.73,
223
+ "learning_rate": 0.0002,
224
+ "loss": 1.3733,
225
+ "step": 360
226
+ },
227
+ {
228
+ "epoch": 0.75,
229
+ "learning_rate": 0.0002,
230
+ "loss": 1.276,
231
+ "step": 370
232
+ },
233
+ {
234
+ "epoch": 0.77,
235
+ "learning_rate": 0.0002,
236
+ "loss": 1.5179,
237
+ "step": 380
238
+ },
239
+ {
240
+ "epoch": 0.79,
241
+ "learning_rate": 0.0002,
242
+ "loss": 1.7899,
243
+ "step": 390
244
+ },
245
+ {
246
+ "epoch": 0.81,
247
+ "learning_rate": 0.0002,
248
+ "loss": 1.8778,
249
+ "step": 400
250
+ },
251
+ {
252
+ "epoch": 0.84,
253
+ "learning_rate": 0.0002,
254
+ "loss": 1.4008,
255
+ "step": 410
256
+ },
257
+ {
258
+ "epoch": 0.86,
259
+ "learning_rate": 0.0002,
260
+ "loss": 1.402,
261
+ "step": 420
262
+ },
263
+ {
264
+ "epoch": 0.88,
265
+ "learning_rate": 0.0002,
266
+ "loss": 1.5183,
267
+ "step": 430
268
+ },
269
+ {
270
+ "epoch": 0.9,
271
+ "learning_rate": 0.0002,
272
+ "loss": 1.8471,
273
+ "step": 440
274
+ },
275
+ {
276
+ "epoch": 0.92,
277
+ "learning_rate": 0.0002,
278
+ "loss": 1.9354,
279
+ "step": 450
280
+ },
281
+ {
282
+ "epoch": 0.94,
283
+ "learning_rate": 0.0002,
284
+ "loss": 1.4035,
285
+ "step": 460
286
+ },
287
+ {
288
+ "epoch": 0.96,
289
+ "learning_rate": 0.0002,
290
+ "loss": 1.4147,
291
+ "step": 470
292
+ },
293
+ {
294
+ "epoch": 0.98,
295
+ "learning_rate": 0.0002,
296
+ "loss": 1.639,
297
+ "step": 480
298
+ },
299
+ {
300
+ "epoch": 1.0,
301
+ "learning_rate": 0.0002,
302
+ "loss": 1.9683,
303
+ "step": 490
304
+ },
305
+ {
306
+ "epoch": 1.02,
307
+ "learning_rate": 0.0002,
308
+ "loss": 1.417,
309
+ "step": 500
310
+ }
311
+ ],
312
+ "logging_steps": 10,
313
+ "max_steps": 500,
314
+ "num_input_tokens_seen": 0,
315
+ "num_train_epochs": 2,
316
+ "save_steps": 50,
317
+ "total_flos": 2.4365101134336e+16,
318
+ "train_batch_size": 2,
319
+ "trial_name": null,
320
+ "trial_params": null
321
+ }
ckpts/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c863c36cb241938d428c573dd639bfd78c550ed3dfefbfd153533898634f7fc0
3
+ size 4728
ckpts/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ torch
2
+ transformers
3
+ accelerate
4
+ einops
5
+ git+https://github.com/huggingface/peft.git