Avnish Kumar commited on
Commit
c9cff18
1 Parent(s): 23af9a5

Upload 11 files

Browse files
README.md CHANGED
@@ -1,23 +1,32 @@
1
  ---
2
- datasets:
3
- - squad
4
- language:
5
- - en
6
- pipeline_tag: question-answering
7
- tags:
8
- - falcon
9
- - llms
10
- - nlp
11
- - ai
12
- - ml
13
- - qlora
14
- - lora
15
- - model quantization
16
- - finetuning
17
- - question and answering
18
  library_name: peft
19
  ---
 
20
 
21
 
 
 
 
 
 
 
 
 
 
 
22
 
23
- Open source large language model - falcon 7b fine tuned for Question and Answering using SQuAD dataset
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  library_name: peft
3
  ---
4
+ ## Training procedure
5
 
6
 
7
+ The following `bitsandbytes` quantization config was used during training:
8
+ - load_in_8bit: False
9
+ - load_in_4bit: True
10
+ - llm_int8_threshold: 6.0
11
+ - llm_int8_skip_modules: None
12
+ - llm_int8_enable_fp32_cpu_offload: False
13
+ - llm_int8_has_fp16_weight: False
14
+ - bnb_4bit_quant_type: nf4
15
+ - bnb_4bit_use_double_quant: False
16
+ - bnb_4bit_compute_dtype: float16
17
 
18
+ The following `bitsandbytes` quantization config was used during training:
19
+ - load_in_8bit: False
20
+ - load_in_4bit: True
21
+ - llm_int8_threshold: 6.0
22
+ - llm_int8_skip_modules: None
23
+ - llm_int8_enable_fp32_cpu_offload: False
24
+ - llm_int8_has_fp16_weight: False
25
+ - bnb_4bit_quant_type: nf4
26
+ - bnb_4bit_use_double_quant: False
27
+ - bnb_4bit_compute_dtype: float16
28
+ ### Framework versions
29
+
30
+ - PEFT 0.4.0.dev0
31
+
32
+ - PEFT 0.4.0.dev0
adapter_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "base_model_name_or_path": "ybelkada/falcon-7b-sharded-bf16",
3
+ "bias": "none",
4
+ "fan_in_fan_out": false,
5
+ "inference_mode": true,
6
+ "init_lora_weights": true,
7
+ "layers_pattern": null,
8
+ "layers_to_transform": null,
9
+ "lora_alpha": 16,
10
+ "lora_dropout": 0.1,
11
+ "modules_to_save": null,
12
+ "peft_type": "LORA",
13
+ "r": 64,
14
+ "revision": null,
15
+ "target_modules": [
16
+ "query_key_value",
17
+ "dense",
18
+ "dense_h_to_4h",
19
+ "dense_4h_to_h"
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:2d93582fc0683aac16e7c793e8ffe464cf5a663b4ff4c1c4c0a5947479017082
3
+ size 522284877
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4cc58f82078b180e931d38d6981b4f1ab3fc9dd048aa72d19e15b4b83383465f
3
+ size 1044539909
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5ca8f2e68e6199ec9cefc400d1ad0abb7d229e688b67c33fb8a3345f13d495a2
3
+ size 14575
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5efc416a6883409dd7ab6f5c779e107c7c2baa7af6e12ed9fbd9dd73b8b20784
3
+ size 627
special_tokens_map.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ ">>TITLE<<",
4
+ ">>ABSTRACT<<",
5
+ ">>INTRODUCTION<<",
6
+ ">>SUMMARY<<",
7
+ ">>COMMENT<<",
8
+ ">>ANSWER<<",
9
+ ">>QUESTION<<",
10
+ ">>DOMAIN<<",
11
+ ">>PREFIX<<",
12
+ ">>SUFFIX<<",
13
+ ">>MIDDLE<<"
14
+ ],
15
+ "eos_token": "<|endoftext|>",
16
+ "pad_token": "<|endoftext|>"
17
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "clean_up_tokenization_spaces": true,
4
+ "eos_token": "<|endoftext|>",
5
+ "model_max_length": 2048,
6
+ "tokenizer_class": "PreTrainedTokenizerFast"
7
+ }
trainer_state.json ADDED
@@ -0,0 +1,316 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 0.20725388601036268,
5
+ "global_step": 500,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.0,
12
+ "learning_rate": 0.0002,
13
+ "loss": 2.2518,
14
+ "step": 10
15
+ },
16
+ {
17
+ "epoch": 0.01,
18
+ "learning_rate": 0.0002,
19
+ "loss": 2.08,
20
+ "step": 20
21
+ },
22
+ {
23
+ "epoch": 0.01,
24
+ "learning_rate": 0.0002,
25
+ "loss": 2.0803,
26
+ "step": 30
27
+ },
28
+ {
29
+ "epoch": 0.02,
30
+ "learning_rate": 0.0002,
31
+ "loss": 2.0389,
32
+ "step": 40
33
+ },
34
+ {
35
+ "epoch": 0.02,
36
+ "learning_rate": 0.0002,
37
+ "loss": 1.9999,
38
+ "step": 50
39
+ },
40
+ {
41
+ "epoch": 0.02,
42
+ "learning_rate": 0.0002,
43
+ "loss": 2.0234,
44
+ "step": 60
45
+ },
46
+ {
47
+ "epoch": 0.03,
48
+ "learning_rate": 0.0002,
49
+ "loss": 2.0212,
50
+ "step": 70
51
+ },
52
+ {
53
+ "epoch": 0.03,
54
+ "learning_rate": 0.0002,
55
+ "loss": 2.0067,
56
+ "step": 80
57
+ },
58
+ {
59
+ "epoch": 0.04,
60
+ "learning_rate": 0.0002,
61
+ "loss": 2.0218,
62
+ "step": 90
63
+ },
64
+ {
65
+ "epoch": 0.04,
66
+ "learning_rate": 0.0002,
67
+ "loss": 1.9615,
68
+ "step": 100
69
+ },
70
+ {
71
+ "epoch": 0.05,
72
+ "learning_rate": 0.0002,
73
+ "loss": 2.0126,
74
+ "step": 110
75
+ },
76
+ {
77
+ "epoch": 0.05,
78
+ "learning_rate": 0.0002,
79
+ "loss": 2.0325,
80
+ "step": 120
81
+ },
82
+ {
83
+ "epoch": 0.05,
84
+ "learning_rate": 0.0002,
85
+ "loss": 1.9627,
86
+ "step": 130
87
+ },
88
+ {
89
+ "epoch": 0.06,
90
+ "learning_rate": 0.0002,
91
+ "loss": 1.9353,
92
+ "step": 140
93
+ },
94
+ {
95
+ "epoch": 0.06,
96
+ "learning_rate": 0.0002,
97
+ "loss": 1.9652,
98
+ "step": 150
99
+ },
100
+ {
101
+ "epoch": 0.07,
102
+ "learning_rate": 0.0002,
103
+ "loss": 2.0244,
104
+ "step": 160
105
+ },
106
+ {
107
+ "epoch": 0.07,
108
+ "learning_rate": 0.0002,
109
+ "loss": 2.0159,
110
+ "step": 170
111
+ },
112
+ {
113
+ "epoch": 0.07,
114
+ "learning_rate": 0.0002,
115
+ "loss": 1.998,
116
+ "step": 180
117
+ },
118
+ {
119
+ "epoch": 0.08,
120
+ "learning_rate": 0.0002,
121
+ "loss": 1.9371,
122
+ "step": 190
123
+ },
124
+ {
125
+ "epoch": 0.08,
126
+ "learning_rate": 0.0002,
127
+ "loss": 1.9921,
128
+ "step": 200
129
+ },
130
+ {
131
+ "epoch": 0.09,
132
+ "learning_rate": 0.0002,
133
+ "loss": 1.9473,
134
+ "step": 210
135
+ },
136
+ {
137
+ "epoch": 0.09,
138
+ "learning_rate": 0.0002,
139
+ "loss": 1.9537,
140
+ "step": 220
141
+ },
142
+ {
143
+ "epoch": 0.1,
144
+ "learning_rate": 0.0002,
145
+ "loss": 2.0283,
146
+ "step": 230
147
+ },
148
+ {
149
+ "epoch": 0.1,
150
+ "learning_rate": 0.0002,
151
+ "loss": 1.9755,
152
+ "step": 240
153
+ },
154
+ {
155
+ "epoch": 0.1,
156
+ "learning_rate": 0.0002,
157
+ "loss": 1.9729,
158
+ "step": 250
159
+ },
160
+ {
161
+ "epoch": 0.11,
162
+ "learning_rate": 0.0002,
163
+ "loss": 2.0241,
164
+ "step": 260
165
+ },
166
+ {
167
+ "epoch": 0.11,
168
+ "learning_rate": 0.0002,
169
+ "loss": 1.9208,
170
+ "step": 270
171
+ },
172
+ {
173
+ "epoch": 0.12,
174
+ "learning_rate": 0.0002,
175
+ "loss": 2.0218,
176
+ "step": 280
177
+ },
178
+ {
179
+ "epoch": 0.12,
180
+ "learning_rate": 0.0002,
181
+ "loss": 1.9854,
182
+ "step": 290
183
+ },
184
+ {
185
+ "epoch": 0.12,
186
+ "learning_rate": 0.0002,
187
+ "loss": 1.9642,
188
+ "step": 300
189
+ },
190
+ {
191
+ "epoch": 0.13,
192
+ "learning_rate": 0.0002,
193
+ "loss": 1.9665,
194
+ "step": 310
195
+ },
196
+ {
197
+ "epoch": 0.13,
198
+ "learning_rate": 0.0002,
199
+ "loss": 1.9638,
200
+ "step": 320
201
+ },
202
+ {
203
+ "epoch": 0.14,
204
+ "learning_rate": 0.0002,
205
+ "loss": 1.9346,
206
+ "step": 330
207
+ },
208
+ {
209
+ "epoch": 0.14,
210
+ "learning_rate": 0.0002,
211
+ "loss": 1.9077,
212
+ "step": 340
213
+ },
214
+ {
215
+ "epoch": 0.15,
216
+ "learning_rate": 0.0002,
217
+ "loss": 1.9674,
218
+ "step": 350
219
+ },
220
+ {
221
+ "epoch": 0.15,
222
+ "learning_rate": 0.0002,
223
+ "loss": 1.9354,
224
+ "step": 360
225
+ },
226
+ {
227
+ "epoch": 0.15,
228
+ "learning_rate": 0.0002,
229
+ "loss": 1.9594,
230
+ "step": 370
231
+ },
232
+ {
233
+ "epoch": 0.16,
234
+ "learning_rate": 0.0002,
235
+ "loss": 1.9699,
236
+ "step": 380
237
+ },
238
+ {
239
+ "epoch": 0.16,
240
+ "learning_rate": 0.0002,
241
+ "loss": 1.9663,
242
+ "step": 390
243
+ },
244
+ {
245
+ "epoch": 0.17,
246
+ "learning_rate": 0.0002,
247
+ "loss": 1.9504,
248
+ "step": 400
249
+ },
250
+ {
251
+ "epoch": 0.17,
252
+ "learning_rate": 0.0002,
253
+ "loss": 1.9155,
254
+ "step": 410
255
+ },
256
+ {
257
+ "epoch": 0.17,
258
+ "learning_rate": 0.0002,
259
+ "loss": 1.8807,
260
+ "step": 420
261
+ },
262
+ {
263
+ "epoch": 0.18,
264
+ "learning_rate": 0.0002,
265
+ "loss": 1.9115,
266
+ "step": 430
267
+ },
268
+ {
269
+ "epoch": 0.18,
270
+ "learning_rate": 0.0002,
271
+ "loss": 1.8983,
272
+ "step": 440
273
+ },
274
+ {
275
+ "epoch": 0.19,
276
+ "learning_rate": 0.0002,
277
+ "loss": 1.8775,
278
+ "step": 450
279
+ },
280
+ {
281
+ "epoch": 0.19,
282
+ "learning_rate": 0.0002,
283
+ "loss": 1.9103,
284
+ "step": 460
285
+ },
286
+ {
287
+ "epoch": 0.19,
288
+ "learning_rate": 0.0002,
289
+ "loss": 1.9505,
290
+ "step": 470
291
+ },
292
+ {
293
+ "epoch": 0.2,
294
+ "learning_rate": 0.0002,
295
+ "loss": 1.9483,
296
+ "step": 480
297
+ },
298
+ {
299
+ "epoch": 0.2,
300
+ "learning_rate": 0.0002,
301
+ "loss": 1.9128,
302
+ "step": 490
303
+ },
304
+ {
305
+ "epoch": 0.21,
306
+ "learning_rate": 0.0002,
307
+ "loss": 1.8843,
308
+ "step": 500
309
+ }
310
+ ],
311
+ "max_steps": 501,
312
+ "num_train_epochs": 1,
313
+ "total_flos": 8.4635406041088e+16,
314
+ "trial_name": null,
315
+ "trial_params": null
316
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d9c770cad0083ffbb1dc2963f4e27123eecf2844cd37a92619601c4803483bd
3
+ size 4027