Pravincoder commited on
Commit
8c40131
1 Parent(s): 763ace8

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - autotrain
4
+ - text-generation
5
+ widget:
6
+ - text: "I love AutoTrain because "
7
+ ---
8
+
9
+ # Model Trained Using AutoTrain
adapter_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "EleutherAI/pythia-1.4b",
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
+ "lora_alpha": 32,
12
+ "lora_dropout": 0.05,
13
+ "modules_to_save": null,
14
+ "peft_type": "LORA",
15
+ "r": 16,
16
+ "rank_pattern": {},
17
+ "revision": null,
18
+ "target_modules": [
19
+ "query_key_value"
20
+ ],
21
+ "task_type": "CAUSAL_LM"
22
+ }
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc14c473ec8bccda37ba5e5234f6b1846f2f7a82a0b90fe196addb795f5cf39c
3
+ size 12600513
checkpoint-2000/README.md ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ ---
4
+ ## Training procedure
5
+
6
+
7
+ The following `bitsandbytes` quantization config was used during training:
8
+ - quant_method: bitsandbytes
9
+ - load_in_8bit: True
10
+ - load_in_4bit: False
11
+ - llm_int8_threshold: 6.0
12
+ - llm_int8_skip_modules: None
13
+ - llm_int8_enable_fp32_cpu_offload: False
14
+ - llm_int8_has_fp16_weight: False
15
+ - bnb_4bit_quant_type: fp4
16
+ - bnb_4bit_use_double_quant: False
17
+ - bnb_4bit_compute_dtype: float32
18
+
19
+ The following `bitsandbytes` quantization config was used during training:
20
+ - quant_method: bitsandbytes
21
+ - load_in_8bit: True
22
+ - load_in_4bit: False
23
+ - llm_int8_threshold: 6.0
24
+ - llm_int8_skip_modules: None
25
+ - llm_int8_enable_fp32_cpu_offload: False
26
+ - llm_int8_has_fp16_weight: False
27
+ - bnb_4bit_quant_type: fp4
28
+ - bnb_4bit_use_double_quant: False
29
+ - bnb_4bit_compute_dtype: float32
30
+ ### Framework versions
31
+
32
+ - PEFT 0.6.0.dev0
33
+
34
+ - PEFT 0.6.0.dev0
checkpoint-2000/adapter_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "EleutherAI/pythia-1.4b",
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
+ "lora_alpha": 32,
12
+ "lora_dropout": 0.05,
13
+ "modules_to_save": null,
14
+ "peft_type": "LORA",
15
+ "r": 16,
16
+ "rank_pattern": {},
17
+ "revision": null,
18
+ "target_modules": [
19
+ "query_key_value"
20
+ ],
21
+ "task_type": "CAUSAL_LM"
22
+ }
checkpoint-2000/adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc14c473ec8bccda37ba5e5234f6b1846f2f7a82a0b90fe196addb795f5cf39c
3
+ size 12600513
checkpoint-2000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1adae34196ec3965e7e84f66e4662daff93788c35046bf9b226b89476f46a75a
3
+ size 25206149
checkpoint-2000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:52f736be3876a5ac9f6410e917bd20e1a67c9ccda046c91a3b5aec1573ae88ba
3
+ size 14511
checkpoint-2000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aafb6a19627506a3cb78949236c86ac9f038aa80fc2f986cbd7c4c03d83a76bc
3
+ size 627
checkpoint-2000/special_tokens_map.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<|endoftext|>",
3
+ "eos_token": "<|endoftext|>",
4
+ "pad_token": "<|endoftext|>",
5
+ "unk_token": "<|endoftext|>"
6
+ }
checkpoint-2000/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-2000/tokenizer_config.json ADDED
@@ -0,0 +1,212 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "<|padding|>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "50254": {
21
+ "content": " ",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": false
27
+ },
28
+ "50255": {
29
+ "content": " ",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": false
35
+ },
36
+ "50256": {
37
+ "content": " ",
38
+ "lstrip": false,
39
+ "normalized": true,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": false
43
+ },
44
+ "50257": {
45
+ "content": " ",
46
+ "lstrip": false,
47
+ "normalized": true,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": false
51
+ },
52
+ "50258": {
53
+ "content": " ",
54
+ "lstrip": false,
55
+ "normalized": true,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": false
59
+ },
60
+ "50259": {
61
+ "content": " ",
62
+ "lstrip": false,
63
+ "normalized": true,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": false
67
+ },
68
+ "50260": {
69
+ "content": " ",
70
+ "lstrip": false,
71
+ "normalized": true,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": false
75
+ },
76
+ "50261": {
77
+ "content": " ",
78
+ "lstrip": false,
79
+ "normalized": true,
80
+ "rstrip": false,
81
+ "single_word": false,
82
+ "special": false
83
+ },
84
+ "50262": {
85
+ "content": " ",
86
+ "lstrip": false,
87
+ "normalized": true,
88
+ "rstrip": false,
89
+ "single_word": false,
90
+ "special": false
91
+ },
92
+ "50263": {
93
+ "content": " ",
94
+ "lstrip": false,
95
+ "normalized": true,
96
+ "rstrip": false,
97
+ "single_word": false,
98
+ "special": false
99
+ },
100
+ "50264": {
101
+ "content": " ",
102
+ "lstrip": false,
103
+ "normalized": true,
104
+ "rstrip": false,
105
+ "single_word": false,
106
+ "special": false
107
+ },
108
+ "50265": {
109
+ "content": " ",
110
+ "lstrip": false,
111
+ "normalized": true,
112
+ "rstrip": false,
113
+ "single_word": false,
114
+ "special": false
115
+ },
116
+ "50266": {
117
+ "content": " ",
118
+ "lstrip": false,
119
+ "normalized": true,
120
+ "rstrip": false,
121
+ "single_word": false,
122
+ "special": false
123
+ },
124
+ "50267": {
125
+ "content": " ",
126
+ "lstrip": false,
127
+ "normalized": true,
128
+ "rstrip": false,
129
+ "single_word": false,
130
+ "special": false
131
+ },
132
+ "50268": {
133
+ "content": " ",
134
+ "lstrip": false,
135
+ "normalized": true,
136
+ "rstrip": false,
137
+ "single_word": false,
138
+ "special": false
139
+ },
140
+ "50269": {
141
+ "content": " ",
142
+ "lstrip": false,
143
+ "normalized": true,
144
+ "rstrip": false,
145
+ "single_word": false,
146
+ "special": false
147
+ },
148
+ "50270": {
149
+ "content": " ",
150
+ "lstrip": false,
151
+ "normalized": true,
152
+ "rstrip": false,
153
+ "single_word": false,
154
+ "special": false
155
+ },
156
+ "50271": {
157
+ "content": " ",
158
+ "lstrip": false,
159
+ "normalized": true,
160
+ "rstrip": false,
161
+ "single_word": false,
162
+ "special": false
163
+ },
164
+ "50272": {
165
+ "content": " ",
166
+ "lstrip": false,
167
+ "normalized": true,
168
+ "rstrip": false,
169
+ "single_word": false,
170
+ "special": false
171
+ },
172
+ "50273": {
173
+ "content": " ",
174
+ "lstrip": false,
175
+ "normalized": true,
176
+ "rstrip": false,
177
+ "single_word": false,
178
+ "special": false
179
+ },
180
+ "50274": {
181
+ "content": " ",
182
+ "lstrip": false,
183
+ "normalized": true,
184
+ "rstrip": false,
185
+ "single_word": false,
186
+ "special": false
187
+ },
188
+ "50275": {
189
+ "content": " ",
190
+ "lstrip": false,
191
+ "normalized": true,
192
+ "rstrip": false,
193
+ "single_word": false,
194
+ "special": false
195
+ },
196
+ "50276": {
197
+ "content": " ",
198
+ "lstrip": false,
199
+ "normalized": true,
200
+ "rstrip": false,
201
+ "single_word": false,
202
+ "special": false
203
+ }
204
+ },
205
+ "additional_special_tokens": [],
206
+ "bos_token": "<|endoftext|>",
207
+ "clean_up_tokenization_spaces": true,
208
+ "eos_token": "<|endoftext|>",
209
+ "model_max_length": 2048,
210
+ "tokenizer_class": "GPTNeoXTokenizer",
211
+ "unk_token": "<|endoftext|>"
212
+ }
checkpoint-2000/trainer_state.json ADDED
@@ -0,0 +1,415 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 199.0662251655629,
5
+ "eval_steps": 500,
6
+ "global_step": 2000,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 2.07,
13
+ "learning_rate": 9.933774834437087e-07,
14
+ "loss": 2.1111,
15
+ "step": 30
16
+ },
17
+ {
18
+ "epoch": 5.07,
19
+ "learning_rate": 1.9867549668874175e-06,
20
+ "loss": 2.0911,
21
+ "step": 60
22
+ },
23
+ {
24
+ "epoch": 8.07,
25
+ "learning_rate": 2.9801324503311258e-06,
26
+ "loss": 2.0488,
27
+ "step": 90
28
+ },
29
+ {
30
+ "epoch": 11.07,
31
+ "learning_rate": 3.973509933774835e-06,
32
+ "loss": 1.9825,
33
+ "step": 120
34
+ },
35
+ {
36
+ "epoch": 14.07,
37
+ "learning_rate": 4.966887417218543e-06,
38
+ "loss": 1.8997,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 17.07,
43
+ "learning_rate": 5.9602649006622515e-06,
44
+ "loss": 1.811,
45
+ "step": 180
46
+ },
47
+ {
48
+ "epoch": 20.07,
49
+ "learning_rate": 6.95364238410596e-06,
50
+ "loss": 1.7265,
51
+ "step": 210
52
+ },
53
+ {
54
+ "epoch": 23.07,
55
+ "learning_rate": 7.94701986754967e-06,
56
+ "loss": 1.6527,
57
+ "step": 240
58
+ },
59
+ {
60
+ "epoch": 26.07,
61
+ "learning_rate": 8.940397350993377e-06,
62
+ "loss": 1.5955,
63
+ "step": 270
64
+ },
65
+ {
66
+ "epoch": 29.07,
67
+ "learning_rate": 9.933774834437086e-06,
68
+ "loss": 1.5492,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 32.07,
73
+ "learning_rate": 1.0927152317880796e-05,
74
+ "loss": 1.508,
75
+ "step": 330
76
+ },
77
+ {
78
+ "epoch": 35.07,
79
+ "learning_rate": 1.1920529801324503e-05,
80
+ "loss": 1.4678,
81
+ "step": 360
82
+ },
83
+ {
84
+ "epoch": 38.07,
85
+ "learning_rate": 1.2913907284768212e-05,
86
+ "loss": 1.4278,
87
+ "step": 390
88
+ },
89
+ {
90
+ "epoch": 41.07,
91
+ "learning_rate": 1.390728476821192e-05,
92
+ "loss": 1.3864,
93
+ "step": 420
94
+ },
95
+ {
96
+ "epoch": 44.07,
97
+ "learning_rate": 1.490066225165563e-05,
98
+ "loss": 1.3418,
99
+ "step": 450
100
+ },
101
+ {
102
+ "epoch": 47.07,
103
+ "learning_rate": 1.589403973509934e-05,
104
+ "loss": 1.2955,
105
+ "step": 480
106
+ },
107
+ {
108
+ "epoch": 50.07,
109
+ "learning_rate": 1.688741721854305e-05,
110
+ "loss": 1.2506,
111
+ "step": 510
112
+ },
113
+ {
114
+ "epoch": 53.07,
115
+ "learning_rate": 1.7880794701986755e-05,
116
+ "loss": 1.2051,
117
+ "step": 540
118
+ },
119
+ {
120
+ "epoch": 56.07,
121
+ "learning_rate": 1.8874172185430464e-05,
122
+ "loss": 1.1617,
123
+ "step": 570
124
+ },
125
+ {
126
+ "epoch": 59.07,
127
+ "learning_rate": 1.9867549668874173e-05,
128
+ "loss": 1.1198,
129
+ "step": 600
130
+ },
131
+ {
132
+ "epoch": 62.07,
133
+ "learning_rate": 2.0860927152317882e-05,
134
+ "loss": 1.0815,
135
+ "step": 630
136
+ },
137
+ {
138
+ "epoch": 65.07,
139
+ "learning_rate": 2.185430463576159e-05,
140
+ "loss": 1.0462,
141
+ "step": 660
142
+ },
143
+ {
144
+ "epoch": 68.07,
145
+ "learning_rate": 2.28476821192053e-05,
146
+ "loss": 1.0112,
147
+ "step": 690
148
+ },
149
+ {
150
+ "epoch": 71.07,
151
+ "learning_rate": 2.3841059602649006e-05,
152
+ "loss": 0.9762,
153
+ "step": 720
154
+ },
155
+ {
156
+ "epoch": 74.07,
157
+ "learning_rate": 2.4834437086092715e-05,
158
+ "loss": 0.941,
159
+ "step": 750
160
+ },
161
+ {
162
+ "epoch": 77.07,
163
+ "learning_rate": 2.5827814569536424e-05,
164
+ "loss": 0.9048,
165
+ "step": 780
166
+ },
167
+ {
168
+ "epoch": 80.07,
169
+ "learning_rate": 2.6821192052980134e-05,
170
+ "loss": 0.8728,
171
+ "step": 810
172
+ },
173
+ {
174
+ "epoch": 83.07,
175
+ "learning_rate": 2.781456953642384e-05,
176
+ "loss": 0.839,
177
+ "step": 840
178
+ },
179
+ {
180
+ "epoch": 86.07,
181
+ "learning_rate": 2.880794701986755e-05,
182
+ "loss": 0.8041,
183
+ "step": 870
184
+ },
185
+ {
186
+ "epoch": 89.07,
187
+ "learning_rate": 2.980132450331126e-05,
188
+ "loss": 0.7714,
189
+ "step": 900
190
+ },
191
+ {
192
+ "epoch": 92.07,
193
+ "learning_rate": 3.079470198675497e-05,
194
+ "loss": 0.7396,
195
+ "step": 930
196
+ },
197
+ {
198
+ "epoch": 95.07,
199
+ "learning_rate": 3.178807947019868e-05,
200
+ "loss": 0.7062,
201
+ "step": 960
202
+ },
203
+ {
204
+ "epoch": 98.07,
205
+ "learning_rate": 3.278145695364239e-05,
206
+ "loss": 0.6731,
207
+ "step": 990
208
+ },
209
+ {
210
+ "epoch": 101.07,
211
+ "learning_rate": 3.37748344370861e-05,
212
+ "loss": 0.6376,
213
+ "step": 1020
214
+ },
215
+ {
216
+ "epoch": 104.07,
217
+ "learning_rate": 3.47682119205298e-05,
218
+ "loss": 0.6083,
219
+ "step": 1050
220
+ },
221
+ {
222
+ "epoch": 107.07,
223
+ "learning_rate": 3.576158940397351e-05,
224
+ "loss": 0.5789,
225
+ "step": 1080
226
+ },
227
+ {
228
+ "epoch": 110.07,
229
+ "learning_rate": 3.675496688741722e-05,
230
+ "loss": 0.5487,
231
+ "step": 1110
232
+ },
233
+ {
234
+ "epoch": 113.07,
235
+ "learning_rate": 3.774834437086093e-05,
236
+ "loss": 0.5202,
237
+ "step": 1140
238
+ },
239
+ {
240
+ "epoch": 116.07,
241
+ "learning_rate": 3.8741721854304637e-05,
242
+ "loss": 0.4941,
243
+ "step": 1170
244
+ },
245
+ {
246
+ "epoch": 119.07,
247
+ "learning_rate": 3.9735099337748346e-05,
248
+ "loss": 0.4675,
249
+ "step": 1200
250
+ },
251
+ {
252
+ "epoch": 122.07,
253
+ "learning_rate": 4.0728476821192055e-05,
254
+ "loss": 0.4421,
255
+ "step": 1230
256
+ },
257
+ {
258
+ "epoch": 125.07,
259
+ "learning_rate": 4.1721854304635764e-05,
260
+ "loss": 0.4175,
261
+ "step": 1260
262
+ },
263
+ {
264
+ "epoch": 128.07,
265
+ "learning_rate": 4.271523178807947e-05,
266
+ "loss": 0.3933,
267
+ "step": 1290
268
+ },
269
+ {
270
+ "epoch": 131.07,
271
+ "learning_rate": 4.370860927152318e-05,
272
+ "loss": 0.3664,
273
+ "step": 1320
274
+ },
275
+ {
276
+ "epoch": 134.07,
277
+ "learning_rate": 4.470198675496689e-05,
278
+ "loss": 0.3446,
279
+ "step": 1350
280
+ },
281
+ {
282
+ "epoch": 137.07,
283
+ "learning_rate": 4.56953642384106e-05,
284
+ "loss": 0.32,
285
+ "step": 1380
286
+ },
287
+ {
288
+ "epoch": 140.07,
289
+ "learning_rate": 4.668874172185431e-05,
290
+ "loss": 0.305,
291
+ "step": 1410
292
+ },
293
+ {
294
+ "epoch": 143.07,
295
+ "learning_rate": 4.768211920529801e-05,
296
+ "loss": 0.2827,
297
+ "step": 1440
298
+ },
299
+ {
300
+ "epoch": 146.07,
301
+ "learning_rate": 4.867549668874172e-05,
302
+ "loss": 0.2646,
303
+ "step": 1470
304
+ },
305
+ {
306
+ "epoch": 149.07,
307
+ "learning_rate": 4.966887417218543e-05,
308
+ "loss": 0.2455,
309
+ "step": 1500
310
+ },
311
+ {
312
+ "epoch": 152.07,
313
+ "learning_rate": 5.0662251655629146e-05,
314
+ "loss": 0.2318,
315
+ "step": 1530
316
+ },
317
+ {
318
+ "epoch": 155.07,
319
+ "learning_rate": 5.165562913907285e-05,
320
+ "loss": 0.2141,
321
+ "step": 1560
322
+ },
323
+ {
324
+ "epoch": 158.07,
325
+ "learning_rate": 5.264900662251656e-05,
326
+ "loss": 0.1944,
327
+ "step": 1590
328
+ },
329
+ {
330
+ "epoch": 161.07,
331
+ "learning_rate": 5.364238410596027e-05,
332
+ "loss": 0.1807,
333
+ "step": 1620
334
+ },
335
+ {
336
+ "epoch": 164.07,
337
+ "learning_rate": 5.4635761589403976e-05,
338
+ "loss": 0.1731,
339
+ "step": 1650
340
+ },
341
+ {
342
+ "epoch": 167.07,
343
+ "learning_rate": 5.562913907284768e-05,
344
+ "loss": 0.1582,
345
+ "step": 1680
346
+ },
347
+ {
348
+ "epoch": 170.07,
349
+ "learning_rate": 5.6622516556291394e-05,
350
+ "loss": 0.1514,
351
+ "step": 1710
352
+ },
353
+ {
354
+ "epoch": 173.07,
355
+ "learning_rate": 5.76158940397351e-05,
356
+ "loss": 0.1379,
357
+ "step": 1740
358
+ },
359
+ {
360
+ "epoch": 176.07,
361
+ "learning_rate": 5.860927152317881e-05,
362
+ "loss": 0.1324,
363
+ "step": 1770
364
+ },
365
+ {
366
+ "epoch": 179.07,
367
+ "learning_rate": 5.960264900662252e-05,
368
+ "loss": 0.1194,
369
+ "step": 1800
370
+ },
371
+ {
372
+ "epoch": 182.07,
373
+ "learning_rate": 6.0596026490066224e-05,
374
+ "loss": 0.1099,
375
+ "step": 1830
376
+ },
377
+ {
378
+ "epoch": 185.07,
379
+ "learning_rate": 6.158940397350994e-05,
380
+ "loss": 0.1051,
381
+ "step": 1860
382
+ },
383
+ {
384
+ "epoch": 188.07,
385
+ "learning_rate": 6.258278145695365e-05,
386
+ "loss": 0.0955,
387
+ "step": 1890
388
+ },
389
+ {
390
+ "epoch": 191.07,
391
+ "learning_rate": 6.357615894039736e-05,
392
+ "loss": 0.094,
393
+ "step": 1920
394
+ },
395
+ {
396
+ "epoch": 194.07,
397
+ "learning_rate": 6.456953642384105e-05,
398
+ "loss": 0.0853,
399
+ "step": 1950
400
+ },
401
+ {
402
+ "epoch": 197.07,
403
+ "learning_rate": 6.556291390728478e-05,
404
+ "loss": 0.0786,
405
+ "step": 1980
406
+ }
407
+ ],
408
+ "logging_steps": 30,
409
+ "max_steps": 30200,
410
+ "num_train_epochs": 200,
411
+ "save_steps": 500,
412
+ "total_flos": 3.2310451765248e+16,
413
+ "trial_name": null,
414
+ "trial_params": null
415
+ }
checkpoint-2000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8877555b98a39dae5880b5c4f790020efbadf582410dafdc23415d644bc37379
3
+ size 4091
config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "EleutherAI/pythia-1.4b",
3
+ "architectures": [
4
+ "GPTNeoXForCausalLM"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": 0.1,
9
+ "eos_token_id": 0,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout": 0.0,
12
+ "hidden_size": 2048,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 8192,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 2048,
17
+ "model_type": "gpt_neox",
18
+ "num_attention_heads": 16,
19
+ "num_hidden_layers": 24,
20
+ "rope_scaling": null,
21
+ "rotary_emb_base": 10000,
22
+ "rotary_pct": 0.25,
23
+ "tie_word_embeddings": false,
24
+ "torch_dtype": "float16",
25
+ "transformers_version": "4.34.0.dev0",
26
+ "use_cache": true,
27
+ "use_parallel_residual": true,
28
+ "vocab_size": 50304
29
+ }
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "eos_token_id": 0,
5
+ "transformers_version": "4.34.0.dev0"
6
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0364d07ff3980a082aa1f8444218644a5a3ec4b6e5a290d3b4c4d0cc65673e0d
3
+ size 2829399677
runs/Oct01_13-06-50_c7315105c6f1/events.out.tfevents.1696165611.c7315105c6f1.2441.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4742e41fb55299886270e924062313999b0abe116cbd11096fcbf93083540c34
3
+ size 15313
special_tokens_map.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<|endoftext|>",
3
+ "eos_token": "<|endoftext|>",
4
+ "unk_token": "<|endoftext|>"
5
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,212 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "<|padding|>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "50254": {
21
+ "content": " ",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": false
27
+ },
28
+ "50255": {
29
+ "content": " ",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": false
35
+ },
36
+ "50256": {
37
+ "content": " ",
38
+ "lstrip": false,
39
+ "normalized": true,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": false
43
+ },
44
+ "50257": {
45
+ "content": " ",
46
+ "lstrip": false,
47
+ "normalized": true,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": false
51
+ },
52
+ "50258": {
53
+ "content": " ",
54
+ "lstrip": false,
55
+ "normalized": true,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": false
59
+ },
60
+ "50259": {
61
+ "content": " ",
62
+ "lstrip": false,
63
+ "normalized": true,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": false
67
+ },
68
+ "50260": {
69
+ "content": " ",
70
+ "lstrip": false,
71
+ "normalized": true,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": false
75
+ },
76
+ "50261": {
77
+ "content": " ",
78
+ "lstrip": false,
79
+ "normalized": true,
80
+ "rstrip": false,
81
+ "single_word": false,
82
+ "special": false
83
+ },
84
+ "50262": {
85
+ "content": " ",
86
+ "lstrip": false,
87
+ "normalized": true,
88
+ "rstrip": false,
89
+ "single_word": false,
90
+ "special": false
91
+ },
92
+ "50263": {
93
+ "content": " ",
94
+ "lstrip": false,
95
+ "normalized": true,
96
+ "rstrip": false,
97
+ "single_word": false,
98
+ "special": false
99
+ },
100
+ "50264": {
101
+ "content": " ",
102
+ "lstrip": false,
103
+ "normalized": true,
104
+ "rstrip": false,
105
+ "single_word": false,
106
+ "special": false
107
+ },
108
+ "50265": {
109
+ "content": " ",
110
+ "lstrip": false,
111
+ "normalized": true,
112
+ "rstrip": false,
113
+ "single_word": false,
114
+ "special": false
115
+ },
116
+ "50266": {
117
+ "content": " ",
118
+ "lstrip": false,
119
+ "normalized": true,
120
+ "rstrip": false,
121
+ "single_word": false,
122
+ "special": false
123
+ },
124
+ "50267": {
125
+ "content": " ",
126
+ "lstrip": false,
127
+ "normalized": true,
128
+ "rstrip": false,
129
+ "single_word": false,
130
+ "special": false
131
+ },
132
+ "50268": {
133
+ "content": " ",
134
+ "lstrip": false,
135
+ "normalized": true,
136
+ "rstrip": false,
137
+ "single_word": false,
138
+ "special": false
139
+ },
140
+ "50269": {
141
+ "content": " ",
142
+ "lstrip": false,
143
+ "normalized": true,
144
+ "rstrip": false,
145
+ "single_word": false,
146
+ "special": false
147
+ },
148
+ "50270": {
149
+ "content": " ",
150
+ "lstrip": false,
151
+ "normalized": true,
152
+ "rstrip": false,
153
+ "single_word": false,
154
+ "special": false
155
+ },
156
+ "50271": {
157
+ "content": " ",
158
+ "lstrip": false,
159
+ "normalized": true,
160
+ "rstrip": false,
161
+ "single_word": false,
162
+ "special": false
163
+ },
164
+ "50272": {
165
+ "content": " ",
166
+ "lstrip": false,
167
+ "normalized": true,
168
+ "rstrip": false,
169
+ "single_word": false,
170
+ "special": false
171
+ },
172
+ "50273": {
173
+ "content": " ",
174
+ "lstrip": false,
175
+ "normalized": true,
176
+ "rstrip": false,
177
+ "single_word": false,
178
+ "special": false
179
+ },
180
+ "50274": {
181
+ "content": " ",
182
+ "lstrip": false,
183
+ "normalized": true,
184
+ "rstrip": false,
185
+ "single_word": false,
186
+ "special": false
187
+ },
188
+ "50275": {
189
+ "content": " ",
190
+ "lstrip": false,
191
+ "normalized": true,
192
+ "rstrip": false,
193
+ "single_word": false,
194
+ "special": false
195
+ },
196
+ "50276": {
197
+ "content": " ",
198
+ "lstrip": false,
199
+ "normalized": true,
200
+ "rstrip": false,
201
+ "single_word": false,
202
+ "special": false
203
+ }
204
+ },
205
+ "additional_special_tokens": [],
206
+ "bos_token": "<|endoftext|>",
207
+ "clean_up_tokenization_spaces": true,
208
+ "eos_token": "<|endoftext|>",
209
+ "model_max_length": 1000000000000000019884624838656,
210
+ "tokenizer_class": "GPTNeoXTokenizer",
211
+ "unk_token": "<|endoftext|>"
212
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8877555b98a39dae5880b5c4f790020efbadf582410dafdc23415d644bc37379
3
+ size 4091
training_params.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"model": "EleutherAI/pythia-1.4b", "data_path": "Pravincoder/Traffic_and_legal_laws", "project_name": "pythia-legal-llm-v4", "train_split": "train", "valid_split": null, "text_column": "text", "lr": 0.0001, "epochs": 200, "batch_size": 2, "warmup_ratio": 0.1, "gradient_accumulation": 1, "optimizer": "adamw_torch", "scheduler": "linear", "weight_decay": 0.0, "max_grad_norm": 1.0, "seed": 42, "add_eos_token": false, "block_size": -1, "use_peft": true, "lora_r": 16, "lora_alpha": 32, "lora_dropout": 0.05, "logging_steps": -1, "evaluation_strategy": "epoch", "save_total_limit": 1, "save_strategy": "epoch", "auto_find_batch_size": true, "fp16": false, "push_to_hub": true, "use_int8": true, "model_max_length": 2048, "repo_id": "Pravincoder/pythia-legal-llm-v4", "use_int4": false, "trainer": "sft", "target_modules": null, "merge_adapter": true, "username": null, "use_flash_attention_2": false}