Shawn Tan commited on
Commit
28d5673
·
1 Parent(s): a8e18b7

Model files.

Browse files
config.json ADDED
@@ -0,0 +1,670 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 0,
3
+ "embedding_dropout": 0,
4
+ "eos_token_id": 0,
5
+ "hidden_size": 1536,
6
+ "init_method": "mup",
7
+ "initializer_range": 0.1,
8
+ "layer_norm_epsilon": 1e-05,
9
+ "m_emb": 12,
10
+ "m_residual": 0.22,
11
+ "m_width": 6,
12
+ "max_position_embeddings": 4096,
13
+ "mlp_blocks": [
14
+ {
15
+ "activation_function": "swiglu",
16
+ "add_bias": false,
17
+ "dropout": 0,
18
+ "intermediate_size": 4096,
19
+ "mlp_type": "MLP"
20
+ },
21
+ {
22
+ "activation_function": "swiglu",
23
+ "add_bias": false,
24
+ "dropout": 0,
25
+ "intermediate_size": 4096,
26
+ "mlp_type": "MLP"
27
+ },
28
+ {
29
+ "activation_function": "swiglu",
30
+ "add_bias": false,
31
+ "dropout": 0,
32
+ "intermediate_size": 4096,
33
+ "mlp_type": "MLP"
34
+ },
35
+ {
36
+ "activation_function": "swiglu",
37
+ "add_bias": false,
38
+ "dropout": 0,
39
+ "intermediate_size": 4096,
40
+ "mlp_type": "MLP"
41
+ },
42
+ {
43
+ "activation_function": "swiglu",
44
+ "add_bias": false,
45
+ "dropout": 0,
46
+ "intermediate_size": 4096,
47
+ "mlp_type": "MLP"
48
+ },
49
+ {
50
+ "activation_function": "swiglu",
51
+ "add_bias": false,
52
+ "dropout": 0,
53
+ "intermediate_size": 4096,
54
+ "mlp_type": "MLP"
55
+ },
56
+ {
57
+ "activation_function": "swiglu",
58
+ "add_bias": false,
59
+ "dropout": 0,
60
+ "intermediate_size": 4096,
61
+ "mlp_type": "MLP"
62
+ },
63
+ {
64
+ "activation_function": "swiglu",
65
+ "add_bias": false,
66
+ "dropout": 0,
67
+ "intermediate_size": 4096,
68
+ "mlp_type": "MLP"
69
+ },
70
+ {
71
+ "activation_function": "swiglu",
72
+ "add_bias": false,
73
+ "dropout": 0,
74
+ "intermediate_size": 4096,
75
+ "mlp_type": "MLP"
76
+ },
77
+ {
78
+ "activation_function": "swiglu",
79
+ "add_bias": false,
80
+ "dropout": 0,
81
+ "intermediate_size": 4096,
82
+ "mlp_type": "MLP"
83
+ },
84
+ {
85
+ "activation_function": "swiglu",
86
+ "add_bias": false,
87
+ "dropout": 0,
88
+ "intermediate_size": 4096,
89
+ "mlp_type": "MLP"
90
+ },
91
+ {
92
+ "activation_function": "swiglu",
93
+ "add_bias": false,
94
+ "dropout": 0,
95
+ "intermediate_size": 4096,
96
+ "mlp_type": "MLP"
97
+ },
98
+ {
99
+ "activation_function": "swiglu",
100
+ "add_bias": false,
101
+ "dropout": 0,
102
+ "intermediate_size": 4096,
103
+ "mlp_type": "MLP"
104
+ },
105
+ {
106
+ "activation_function": "swiglu",
107
+ "add_bias": false,
108
+ "dropout": 0,
109
+ "intermediate_size": 4096,
110
+ "mlp_type": "MLP"
111
+ },
112
+ {
113
+ "activation_function": "swiglu",
114
+ "add_bias": false,
115
+ "dropout": 0,
116
+ "intermediate_size": 4096,
117
+ "mlp_type": "MLP"
118
+ },
119
+ {
120
+ "activation_function": "swiglu",
121
+ "add_bias": false,
122
+ "dropout": 0,
123
+ "intermediate_size": 4096,
124
+ "mlp_type": "MLP"
125
+ },
126
+ {
127
+ "activation_function": "swiglu",
128
+ "add_bias": false,
129
+ "dropout": 0,
130
+ "intermediate_size": 4096,
131
+ "mlp_type": "MLP"
132
+ },
133
+ {
134
+ "activation_function": "swiglu",
135
+ "add_bias": false,
136
+ "dropout": 0,
137
+ "intermediate_size": 4096,
138
+ "mlp_type": "MLP"
139
+ },
140
+ {
141
+ "activation_function": "swiglu",
142
+ "add_bias": false,
143
+ "dropout": 0,
144
+ "intermediate_size": 4096,
145
+ "mlp_type": "MLP"
146
+ },
147
+ {
148
+ "activation_function": "swiglu",
149
+ "add_bias": false,
150
+ "dropout": 0,
151
+ "intermediate_size": 4096,
152
+ "mlp_type": "MLP"
153
+ },
154
+ {
155
+ "activation_function": "swiglu",
156
+ "add_bias": false,
157
+ "dropout": 0,
158
+ "intermediate_size": 4096,
159
+ "mlp_type": "MLP"
160
+ },
161
+ {
162
+ "activation_function": "swiglu",
163
+ "add_bias": false,
164
+ "dropout": 0,
165
+ "intermediate_size": 4096,
166
+ "mlp_type": "MLP"
167
+ },
168
+ {
169
+ "activation_function": "swiglu",
170
+ "add_bias": false,
171
+ "dropout": 0,
172
+ "intermediate_size": 4096,
173
+ "mlp_type": "MLP"
174
+ },
175
+ {
176
+ "activation_function": "swiglu",
177
+ "add_bias": false,
178
+ "dropout": 0,
179
+ "intermediate_size": 4096,
180
+ "mlp_type": "MLP"
181
+ },
182
+ {
183
+ "activation_function": "swiglu",
184
+ "add_bias": false,
185
+ "dropout": 0,
186
+ "intermediate_size": 4096,
187
+ "mlp_type": "MLP"
188
+ },
189
+ {
190
+ "activation_function": "swiglu",
191
+ "add_bias": false,
192
+ "dropout": 0,
193
+ "intermediate_size": 4096,
194
+ "mlp_type": "MLP"
195
+ },
196
+ {
197
+ "activation_function": "swiglu",
198
+ "add_bias": false,
199
+ "dropout": 0,
200
+ "intermediate_size": 4096,
201
+ "mlp_type": "MLP"
202
+ },
203
+ {
204
+ "activation_function": "swiglu",
205
+ "add_bias": false,
206
+ "dropout": 0,
207
+ "intermediate_size": 4096,
208
+ "mlp_type": "MLP"
209
+ },
210
+ {
211
+ "activation_function": "swiglu",
212
+ "add_bias": false,
213
+ "dropout": 0,
214
+ "intermediate_size": 4096,
215
+ "mlp_type": "MLP"
216
+ },
217
+ {
218
+ "activation_function": "swiglu",
219
+ "add_bias": false,
220
+ "dropout": 0,
221
+ "intermediate_size": 4096,
222
+ "mlp_type": "MLP"
223
+ },
224
+ {
225
+ "activation_function": "swiglu",
226
+ "add_bias": false,
227
+ "dropout": 0,
228
+ "intermediate_size": 4096,
229
+ "mlp_type": "MLP"
230
+ },
231
+ {
232
+ "activation_function": "swiglu",
233
+ "add_bias": false,
234
+ "dropout": 0,
235
+ "intermediate_size": 4096,
236
+ "mlp_type": "MLP"
237
+ },
238
+ {
239
+ "activation_function": "swiglu",
240
+ "add_bias": false,
241
+ "dropout": 0,
242
+ "intermediate_size": 4096,
243
+ "mlp_type": "MLP"
244
+ },
245
+ {
246
+ "activation_function": "swiglu",
247
+ "add_bias": false,
248
+ "dropout": 0,
249
+ "intermediate_size": 4096,
250
+ "mlp_type": "MLP"
251
+ },
252
+ {
253
+ "activation_function": "swiglu",
254
+ "add_bias": false,
255
+ "dropout": 0,
256
+ "intermediate_size": 4096,
257
+ "mlp_type": "MLP"
258
+ },
259
+ {
260
+ "activation_function": "swiglu",
261
+ "add_bias": false,
262
+ "dropout": 0,
263
+ "intermediate_size": 4096,
264
+ "mlp_type": "MLP"
265
+ },
266
+ {
267
+ "activation_function": "swiglu",
268
+ "add_bias": false,
269
+ "dropout": 0,
270
+ "intermediate_size": 4096,
271
+ "mlp_type": "MLP"
272
+ },
273
+ {
274
+ "activation_function": "swiglu",
275
+ "add_bias": false,
276
+ "dropout": 0,
277
+ "intermediate_size": 4096,
278
+ "mlp_type": "MLP"
279
+ },
280
+ {
281
+ "activation_function": "swiglu",
282
+ "add_bias": false,
283
+ "dropout": 0,
284
+ "intermediate_size": 4096,
285
+ "mlp_type": "MLP"
286
+ },
287
+ {
288
+ "activation_function": "swiglu",
289
+ "add_bias": false,
290
+ "dropout": 0,
291
+ "intermediate_size": 4096,
292
+ "mlp_type": "MLP"
293
+ }
294
+ ],
295
+ "model_type": "gpt_dolomite",
296
+ "normalization_function": "rmsnorm",
297
+ "num_attention_heads": 12,
298
+ "num_layers": 40,
299
+ "pad_token_id": 0,
300
+ "position_embedding_type": "nope",
301
+ "rope_scaling": null,
302
+ "rope_theta": 10000,
303
+ "router_aux_loss_coef": 0.01,
304
+ "sequence_mixer_blocks": [
305
+ {
306
+ "add_bias": false,
307
+ "attention_head_type": "gqa",
308
+ "attention_multiplier": 0.0078125,
309
+ "dropout": 0,
310
+ "num_key_value_heads": 4,
311
+ "sequence_mixer_type": "stickbreaking_attention",
312
+ "softmax_dropout": 0
313
+ },
314
+ {
315
+ "add_bias": false,
316
+ "attention_head_type": "gqa",
317
+ "attention_multiplier": 0.0078125,
318
+ "dropout": 0,
319
+ "num_key_value_heads": 4,
320
+ "sequence_mixer_type": "stickbreaking_attention",
321
+ "softmax_dropout": 0
322
+ },
323
+ {
324
+ "add_bias": false,
325
+ "attention_head_type": "gqa",
326
+ "attention_multiplier": 0.0078125,
327
+ "dropout": 0,
328
+ "num_key_value_heads": 4,
329
+ "sequence_mixer_type": "stickbreaking_attention",
330
+ "softmax_dropout": 0
331
+ },
332
+ {
333
+ "add_bias": false,
334
+ "attention_head_type": "gqa",
335
+ "attention_multiplier": 0.0078125,
336
+ "dropout": 0,
337
+ "num_key_value_heads": 4,
338
+ "sequence_mixer_type": "stickbreaking_attention",
339
+ "softmax_dropout": 0
340
+ },
341
+ {
342
+ "add_bias": false,
343
+ "attention_head_type": "gqa",
344
+ "attention_multiplier": 0.0078125,
345
+ "dropout": 0,
346
+ "num_key_value_heads": 4,
347
+ "sequence_mixer_type": "stickbreaking_attention",
348
+ "softmax_dropout": 0
349
+ },
350
+ {
351
+ "add_bias": false,
352
+ "attention_head_type": "gqa",
353
+ "attention_multiplier": 0.0078125,
354
+ "dropout": 0,
355
+ "num_key_value_heads": 4,
356
+ "sequence_mixer_type": "stickbreaking_attention",
357
+ "softmax_dropout": 0
358
+ },
359
+ {
360
+ "add_bias": false,
361
+ "attention_head_type": "gqa",
362
+ "attention_multiplier": 0.0078125,
363
+ "dropout": 0,
364
+ "num_key_value_heads": 4,
365
+ "sequence_mixer_type": "stickbreaking_attention",
366
+ "softmax_dropout": 0
367
+ },
368
+ {
369
+ "add_bias": false,
370
+ "attention_head_type": "gqa",
371
+ "attention_multiplier": 0.0078125,
372
+ "dropout": 0,
373
+ "num_key_value_heads": 4,
374
+ "sequence_mixer_type": "stickbreaking_attention",
375
+ "softmax_dropout": 0
376
+ },
377
+ {
378
+ "add_bias": false,
379
+ "attention_head_type": "gqa",
380
+ "attention_multiplier": 0.0078125,
381
+ "dropout": 0,
382
+ "num_key_value_heads": 4,
383
+ "sequence_mixer_type": "stickbreaking_attention",
384
+ "softmax_dropout": 0
385
+ },
386
+ {
387
+ "add_bias": false,
388
+ "attention_head_type": "gqa",
389
+ "attention_multiplier": 0.0078125,
390
+ "dropout": 0,
391
+ "num_key_value_heads": 4,
392
+ "sequence_mixer_type": "stickbreaking_attention",
393
+ "softmax_dropout": 0
394
+ },
395
+ {
396
+ "add_bias": false,
397
+ "attention_head_type": "gqa",
398
+ "attention_multiplier": 0.0078125,
399
+ "dropout": 0,
400
+ "num_key_value_heads": 4,
401
+ "sequence_mixer_type": "stickbreaking_attention",
402
+ "softmax_dropout": 0
403
+ },
404
+ {
405
+ "add_bias": false,
406
+ "attention_head_type": "gqa",
407
+ "attention_multiplier": 0.0078125,
408
+ "dropout": 0,
409
+ "num_key_value_heads": 4,
410
+ "sequence_mixer_type": "stickbreaking_attention",
411
+ "softmax_dropout": 0
412
+ },
413
+ {
414
+ "add_bias": false,
415
+ "attention_head_type": "gqa",
416
+ "attention_multiplier": 0.0078125,
417
+ "dropout": 0,
418
+ "num_key_value_heads": 4,
419
+ "sequence_mixer_type": "stickbreaking_attention",
420
+ "softmax_dropout": 0
421
+ },
422
+ {
423
+ "add_bias": false,
424
+ "attention_head_type": "gqa",
425
+ "attention_multiplier": 0.0078125,
426
+ "dropout": 0,
427
+ "num_key_value_heads": 4,
428
+ "sequence_mixer_type": "stickbreaking_attention",
429
+ "softmax_dropout": 0
430
+ },
431
+ {
432
+ "add_bias": false,
433
+ "attention_head_type": "gqa",
434
+ "attention_multiplier": 0.0078125,
435
+ "dropout": 0,
436
+ "num_key_value_heads": 4,
437
+ "sequence_mixer_type": "stickbreaking_attention",
438
+ "softmax_dropout": 0
439
+ },
440
+ {
441
+ "add_bias": false,
442
+ "attention_head_type": "gqa",
443
+ "attention_multiplier": 0.0078125,
444
+ "dropout": 0,
445
+ "num_key_value_heads": 4,
446
+ "sequence_mixer_type": "stickbreaking_attention",
447
+ "softmax_dropout": 0
448
+ },
449
+ {
450
+ "add_bias": false,
451
+ "attention_head_type": "gqa",
452
+ "attention_multiplier": 0.0078125,
453
+ "dropout": 0,
454
+ "num_key_value_heads": 4,
455
+ "sequence_mixer_type": "stickbreaking_attention",
456
+ "softmax_dropout": 0
457
+ },
458
+ {
459
+ "add_bias": false,
460
+ "attention_head_type": "gqa",
461
+ "attention_multiplier": 0.0078125,
462
+ "dropout": 0,
463
+ "num_key_value_heads": 4,
464
+ "sequence_mixer_type": "stickbreaking_attention",
465
+ "softmax_dropout": 0
466
+ },
467
+ {
468
+ "add_bias": false,
469
+ "attention_head_type": "gqa",
470
+ "attention_multiplier": 0.0078125,
471
+ "dropout": 0,
472
+ "num_key_value_heads": 4,
473
+ "sequence_mixer_type": "stickbreaking_attention",
474
+ "softmax_dropout": 0
475
+ },
476
+ {
477
+ "add_bias": false,
478
+ "attention_head_type": "gqa",
479
+ "attention_multiplier": 0.0078125,
480
+ "dropout": 0,
481
+ "num_key_value_heads": 4,
482
+ "sequence_mixer_type": "stickbreaking_attention",
483
+ "softmax_dropout": 0
484
+ },
485
+ {
486
+ "add_bias": false,
487
+ "attention_head_type": "gqa",
488
+ "attention_multiplier": 0.0078125,
489
+ "dropout": 0,
490
+ "num_key_value_heads": 4,
491
+ "sequence_mixer_type": "stickbreaking_attention",
492
+ "softmax_dropout": 0
493
+ },
494
+ {
495
+ "add_bias": false,
496
+ "attention_head_type": "gqa",
497
+ "attention_multiplier": 0.0078125,
498
+ "dropout": 0,
499
+ "num_key_value_heads": 4,
500
+ "sequence_mixer_type": "stickbreaking_attention",
501
+ "softmax_dropout": 0
502
+ },
503
+ {
504
+ "add_bias": false,
505
+ "attention_head_type": "gqa",
506
+ "attention_multiplier": 0.0078125,
507
+ "dropout": 0,
508
+ "num_key_value_heads": 4,
509
+ "sequence_mixer_type": "stickbreaking_attention",
510
+ "softmax_dropout": 0
511
+ },
512
+ {
513
+ "add_bias": false,
514
+ "attention_head_type": "gqa",
515
+ "attention_multiplier": 0.0078125,
516
+ "dropout": 0,
517
+ "num_key_value_heads": 4,
518
+ "sequence_mixer_type": "stickbreaking_attention",
519
+ "softmax_dropout": 0
520
+ },
521
+ {
522
+ "add_bias": false,
523
+ "attention_head_type": "gqa",
524
+ "attention_multiplier": 0.0078125,
525
+ "dropout": 0,
526
+ "num_key_value_heads": 4,
527
+ "sequence_mixer_type": "stickbreaking_attention",
528
+ "softmax_dropout": 0
529
+ },
530
+ {
531
+ "add_bias": false,
532
+ "attention_head_type": "gqa",
533
+ "attention_multiplier": 0.0078125,
534
+ "dropout": 0,
535
+ "num_key_value_heads": 4,
536
+ "sequence_mixer_type": "stickbreaking_attention",
537
+ "softmax_dropout": 0
538
+ },
539
+ {
540
+ "add_bias": false,
541
+ "attention_head_type": "gqa",
542
+ "attention_multiplier": 0.0078125,
543
+ "dropout": 0,
544
+ "num_key_value_heads": 4,
545
+ "sequence_mixer_type": "stickbreaking_attention",
546
+ "softmax_dropout": 0
547
+ },
548
+ {
549
+ "add_bias": false,
550
+ "attention_head_type": "gqa",
551
+ "attention_multiplier": 0.0078125,
552
+ "dropout": 0,
553
+ "num_key_value_heads": 4,
554
+ "sequence_mixer_type": "stickbreaking_attention",
555
+ "softmax_dropout": 0
556
+ },
557
+ {
558
+ "add_bias": false,
559
+ "attention_head_type": "gqa",
560
+ "attention_multiplier": 0.0078125,
561
+ "dropout": 0,
562
+ "num_key_value_heads": 4,
563
+ "sequence_mixer_type": "stickbreaking_attention",
564
+ "softmax_dropout": 0
565
+ },
566
+ {
567
+ "add_bias": false,
568
+ "attention_head_type": "gqa",
569
+ "attention_multiplier": 0.0078125,
570
+ "dropout": 0,
571
+ "num_key_value_heads": 4,
572
+ "sequence_mixer_type": "stickbreaking_attention",
573
+ "softmax_dropout": 0
574
+ },
575
+ {
576
+ "add_bias": false,
577
+ "attention_head_type": "gqa",
578
+ "attention_multiplier": 0.0078125,
579
+ "dropout": 0,
580
+ "num_key_value_heads": 4,
581
+ "sequence_mixer_type": "stickbreaking_attention",
582
+ "softmax_dropout": 0
583
+ },
584
+ {
585
+ "add_bias": false,
586
+ "attention_head_type": "gqa",
587
+ "attention_multiplier": 0.0078125,
588
+ "dropout": 0,
589
+ "num_key_value_heads": 4,
590
+ "sequence_mixer_type": "stickbreaking_attention",
591
+ "softmax_dropout": 0
592
+ },
593
+ {
594
+ "add_bias": false,
595
+ "attention_head_type": "gqa",
596
+ "attention_multiplier": 0.0078125,
597
+ "dropout": 0,
598
+ "num_key_value_heads": 4,
599
+ "sequence_mixer_type": "stickbreaking_attention",
600
+ "softmax_dropout": 0
601
+ },
602
+ {
603
+ "add_bias": false,
604
+ "attention_head_type": "gqa",
605
+ "attention_multiplier": 0.0078125,
606
+ "dropout": 0,
607
+ "num_key_value_heads": 4,
608
+ "sequence_mixer_type": "stickbreaking_attention",
609
+ "softmax_dropout": 0
610
+ },
611
+ {
612
+ "add_bias": false,
613
+ "attention_head_type": "gqa",
614
+ "attention_multiplier": 0.0078125,
615
+ "dropout": 0,
616
+ "num_key_value_heads": 4,
617
+ "sequence_mixer_type": "stickbreaking_attention",
618
+ "softmax_dropout": 0
619
+ },
620
+ {
621
+ "add_bias": false,
622
+ "attention_head_type": "gqa",
623
+ "attention_multiplier": 0.0078125,
624
+ "dropout": 0,
625
+ "num_key_value_heads": 4,
626
+ "sequence_mixer_type": "stickbreaking_attention",
627
+ "softmax_dropout": 0
628
+ },
629
+ {
630
+ "add_bias": false,
631
+ "attention_head_type": "gqa",
632
+ "attention_multiplier": 0.0078125,
633
+ "dropout": 0,
634
+ "num_key_value_heads": 4,
635
+ "sequence_mixer_type": "stickbreaking_attention",
636
+ "softmax_dropout": 0
637
+ },
638
+ {
639
+ "add_bias": false,
640
+ "attention_head_type": "gqa",
641
+ "attention_multiplier": 0.0078125,
642
+ "dropout": 0,
643
+ "num_key_value_heads": 4,
644
+ "sequence_mixer_type": "stickbreaking_attention",
645
+ "softmax_dropout": 0
646
+ },
647
+ {
648
+ "add_bias": false,
649
+ "attention_head_type": "gqa",
650
+ "attention_multiplier": 0.0078125,
651
+ "dropout": 0,
652
+ "num_key_value_heads": 4,
653
+ "sequence_mixer_type": "stickbreaking_attention",
654
+ "softmax_dropout": 0
655
+ },
656
+ {
657
+ "add_bias": false,
658
+ "attention_head_type": "gqa",
659
+ "attention_multiplier": 0.0078125,
660
+ "dropout": 0,
661
+ "num_key_value_heads": 4,
662
+ "sequence_mixer_type": "stickbreaking_attention",
663
+ "softmax_dropout": 0
664
+ }
665
+ ],
666
+ "transformers_version": "4.49.0.dev0",
667
+ "upcast_logits_for_loss": false,
668
+ "use_cache": true,
669
+ "vocab_size": 50304
670
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ec0af7b12d74ccf67df9fe244344190ba4060b5aaa3fd16495c44c68cd48ad8
3
+ size 4336875624
special_tokens_map.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|endoftext|>",
4
+ "<fim_prefix>",
5
+ "<fim_middle>",
6
+ "<fim_suffix>",
7
+ "<fim_pad>",
8
+ "<filename>",
9
+ "<gh_stars>",
10
+ "<issue_start>",
11
+ "<issue_comment>",
12
+ "<issue_closed>",
13
+ "<jupyter_start>",
14
+ "<jupyter_text>",
15
+ "<jupyter_code>",
16
+ "<jupyter_output>",
17
+ "<empty_output>",
18
+ "<commit_before>",
19
+ "<commit_msg>",
20
+ "<commit_after>",
21
+ "<reponame>"
22
+ ],
23
+ "bos_token": {
24
+ "content": "<|endoftext|>",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "eos_token": {
31
+ "content": "<|endoftext|>",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "unk_token": {
38
+ "content": "<|endoftext|>",
39
+ "lstrip": false,
40
+ "normalized": false,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ }
44
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,186 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<|endoftext|>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<fim_prefix>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "<fim_middle>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<fim_suffix>",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "4": {
37
+ "content": "<fim_pad>",
38
+ "lstrip": false,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ },
44
+ "5": {
45
+ "content": "<filename>",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": true
51
+ },
52
+ "6": {
53
+ "content": "<gh_stars>",
54
+ "lstrip": false,
55
+ "normalized": false,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": true
59
+ },
60
+ "7": {
61
+ "content": "<issue_start>",
62
+ "lstrip": false,
63
+ "normalized": false,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": true
67
+ },
68
+ "8": {
69
+ "content": "<issue_comment>",
70
+ "lstrip": false,
71
+ "normalized": false,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": true
75
+ },
76
+ "9": {
77
+ "content": "<issue_closed>",
78
+ "lstrip": false,
79
+ "normalized": false,
80
+ "rstrip": false,
81
+ "single_word": false,
82
+ "special": true
83
+ },
84
+ "10": {
85
+ "content": "<jupyter_start>",
86
+ "lstrip": false,
87
+ "normalized": false,
88
+ "rstrip": false,
89
+ "single_word": false,
90
+ "special": true
91
+ },
92
+ "11": {
93
+ "content": "<jupyter_text>",
94
+ "lstrip": false,
95
+ "normalized": false,
96
+ "rstrip": false,
97
+ "single_word": false,
98
+ "special": true
99
+ },
100
+ "12": {
101
+ "content": "<jupyter_code>",
102
+ "lstrip": false,
103
+ "normalized": false,
104
+ "rstrip": false,
105
+ "single_word": false,
106
+ "special": true
107
+ },
108
+ "13": {
109
+ "content": "<jupyter_output>",
110
+ "lstrip": false,
111
+ "normalized": false,
112
+ "rstrip": false,
113
+ "single_word": false,
114
+ "special": true
115
+ },
116
+ "14": {
117
+ "content": "<empty_output>",
118
+ "lstrip": false,
119
+ "normalized": false,
120
+ "rstrip": false,
121
+ "single_word": false,
122
+ "special": true
123
+ },
124
+ "15": {
125
+ "content": "<commit_before>",
126
+ "lstrip": false,
127
+ "normalized": false,
128
+ "rstrip": false,
129
+ "single_word": false,
130
+ "special": true
131
+ },
132
+ "16": {
133
+ "content": "<commit_msg>",
134
+ "lstrip": false,
135
+ "normalized": false,
136
+ "rstrip": false,
137
+ "single_word": false,
138
+ "special": true
139
+ },
140
+ "17": {
141
+ "content": "<commit_after>",
142
+ "lstrip": false,
143
+ "normalized": false,
144
+ "rstrip": false,
145
+ "single_word": false,
146
+ "special": true
147
+ },
148
+ "18": {
149
+ "content": "<reponame>",
150
+ "lstrip": false,
151
+ "normalized": false,
152
+ "rstrip": false,
153
+ "single_word": false,
154
+ "special": true
155
+ }
156
+ },
157
+ "additional_special_tokens": [
158
+ "<|endoftext|>",
159
+ "<fim_prefix>",
160
+ "<fim_middle>",
161
+ "<fim_suffix>",
162
+ "<fim_pad>",
163
+ "<filename>",
164
+ "<gh_stars>",
165
+ "<issue_start>",
166
+ "<issue_comment>",
167
+ "<issue_closed>",
168
+ "<jupyter_start>",
169
+ "<jupyter_text>",
170
+ "<jupyter_code>",
171
+ "<jupyter_output>",
172
+ "<empty_output>",
173
+ "<commit_before>",
174
+ "<commit_msg>",
175
+ "<commit_after>",
176
+ "<reponame>"
177
+ ],
178
+ "bos_token": "<|endoftext|>",
179
+ "clean_up_tokenization_spaces": false,
180
+ "eos_token": "<|endoftext|>",
181
+ "extra_special_tokens": {},
182
+ "model_max_length": 1000000000000000019884624838656,
183
+ "tokenizer_class": "GPT2Tokenizer",
184
+ "unk_token": "<|endoftext|>",
185
+ "vocab_size": 49152
186
+ }