tomjennings100 commited on
Commit
5f247d5
·
1 Parent(s): 5157ecd

add model

Browse files
README.md CHANGED
@@ -1,4 +1,78 @@
1
- mistral 7b finetuned on curation-corpu
2
  ---
3
- license: mit
 
 
 
 
 
 
4
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ license: apache-2.0
3
+ base_model: mistralai/Mistral-7B-v0.1
4
+ tags:
5
+ - generated_from_trainer
6
+ model-index:
7
+ - name: qlora-out
8
+ results: []
9
  ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ [<img src="https://raw.githubusercontent.com/OpenAccess-AI-Collective/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/OpenAccess-AI-Collective/axolotl)
15
+ # qlora-out
16
+
17
+ This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on the None dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 1.2423
20
+
21
+ ## Model description
22
+
23
+ More information needed
24
+
25
+ ## Intended uses & limitations
26
+
27
+ More information needed
28
+
29
+ ## Training and evaluation data
30
+
31
+ More information needed
32
+
33
+ ## Training procedure
34
+
35
+ ### Training hyperparameters
36
+
37
+ The following hyperparameters were used during training:
38
+ - learning_rate: 0.0002
39
+ - train_batch_size: 2
40
+ - eval_batch_size: 2
41
+ - seed: 42
42
+ - gradient_accumulation_steps: 4
43
+ - total_train_batch_size: 8
44
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
+ - lr_scheduler_type: cosine
46
+ - lr_scheduler_warmup_steps: 10
47
+ - num_epochs: 1
48
+
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Validation Loss |
52
+ |:-------------:|:-----:|:----:|:---------------:|
53
+ | 1.5123 | 0.01 | 1 | 1.5038 |
54
+ | 1.3662 | 0.06 | 5 | 1.4103 |
55
+ | 1.1836 | 0.11 | 10 | 1.3055 |
56
+ | 1.2761 | 0.17 | 15 | 1.2810 |
57
+ | 1.1779 | 0.22 | 20 | 1.2696 |
58
+ | 1.1242 | 0.28 | 25 | 1.2642 |
59
+ | 1.2414 | 0.33 | 30 | 1.2588 |
60
+ | 1.1382 | 0.39 | 35 | 1.2555 |
61
+ | 1.2094 | 0.45 | 40 | 1.2520 |
62
+ | 1.1049 | 0.5 | 45 | 1.2504 |
63
+ | 1.1709 | 0.56 | 50 | 1.2487 |
64
+ | 1.0981 | 0.61 | 55 | 1.2463 |
65
+ | 1.1902 | 0.67 | 60 | 1.2446 |
66
+ | 1.1526 | 0.72 | 65 | 1.2446 |
67
+ | 1.1319 | 0.78 | 70 | 1.2440 |
68
+ | 1.1913 | 0.84 | 75 | 1.2430 |
69
+ | 1.1875 | 0.89 | 80 | 1.2424 |
70
+ | 1.1454 | 0.95 | 85 | 1.2423 |
71
+
72
+
73
+ ### Framework versions
74
+
75
+ - Transformers 4.35.2
76
+ - Pytorch 2.0.1+cu118
77
+ - Datasets 2.15.0
78
+ - Tokenizers 0.15.0
adapter_config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "mistralai/Mistral-7B-v0.1",
5
+ "bias": "none",
6
+ "fan_in_fan_out": null,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layers_pattern": null,
10
+ "layers_to_transform": null,
11
+ "lora_alpha": 16,
12
+ "lora_dropout": 0.05,
13
+ "modules_to_save": null,
14
+ "peft_type": "LORA",
15
+ "r": 32,
16
+ "rank_pattern": {},
17
+ "revision": null,
18
+ "target_modules": [
19
+ "o_proj",
20
+ "down_proj",
21
+ "k_proj",
22
+ "up_proj",
23
+ "v_proj",
24
+ "gate_proj",
25
+ "q_proj"
26
+ ],
27
+ "task_type": "CAUSAL_LM"
28
+ }
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9bb626358ee09186105c032bf3f6ffd6106eceed582ef9adce9ab30530e440f5
3
+ size 335705741
checkpoint-89/README.md ADDED
@@ -0,0 +1,220 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ base_model: mistralai/Mistral-7B-v0.1
4
+ ---
5
+
6
+ # Model Card for Model ID
7
+
8
+ <!-- Provide a quick summary of what the model is/does. -->
9
+
10
+
11
+
12
+ ## Model Details
13
+
14
+ ### Model Description
15
+
16
+ <!-- Provide a longer summary of what this model is. -->
17
+
18
+
19
+
20
+ - **Developed by:** [More Information Needed]
21
+ - **Funded by [optional]:** [More Information Needed]
22
+ - **Shared by [optional]:** [More Information Needed]
23
+ - **Model type:** [More Information Needed]
24
+ - **Language(s) (NLP):** [More Information Needed]
25
+ - **License:** [More Information Needed]
26
+ - **Finetuned from model [optional]:** [More Information Needed]
27
+
28
+ ### Model Sources [optional]
29
+
30
+ <!-- Provide the basic links for the model. -->
31
+
32
+ - **Repository:** [More Information Needed]
33
+ - **Paper [optional]:** [More Information Needed]
34
+ - **Demo [optional]:** [More Information Needed]
35
+
36
+ ## Uses
37
+
38
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
39
+
40
+ ### Direct Use
41
+
42
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
43
+
44
+ [More Information Needed]
45
+
46
+ ### Downstream Use [optional]
47
+
48
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
49
+
50
+ [More Information Needed]
51
+
52
+ ### Out-of-Scope Use
53
+
54
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
55
+
56
+ [More Information Needed]
57
+
58
+ ## Bias, Risks, and Limitations
59
+
60
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
61
+
62
+ [More Information Needed]
63
+
64
+ ### Recommendations
65
+
66
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
67
+
68
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
69
+
70
+ ## How to Get Started with the Model
71
+
72
+ Use the code below to get started with the model.
73
+
74
+ [More Information Needed]
75
+
76
+ ## Training Details
77
+
78
+ ### Training Data
79
+
80
+ <!-- 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. -->
81
+
82
+ [More Information Needed]
83
+
84
+ ### Training Procedure
85
+
86
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
87
+
88
+ #### Preprocessing [optional]
89
+
90
+ [More Information Needed]
91
+
92
+
93
+ #### Training Hyperparameters
94
+
95
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
96
+
97
+ #### Speeds, Sizes, Times [optional]
98
+
99
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
100
+
101
+ [More Information Needed]
102
+
103
+ ## Evaluation
104
+
105
+ <!-- This section describes the evaluation protocols and provides the results. -->
106
+
107
+ ### Testing Data, Factors & Metrics
108
+
109
+ #### Testing Data
110
+
111
+ <!-- This should link to a Dataset Card if possible. -->
112
+
113
+ [More Information Needed]
114
+
115
+ #### Factors
116
+
117
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
118
+
119
+ [More Information Needed]
120
+
121
+ #### Metrics
122
+
123
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
124
+
125
+ [More Information Needed]
126
+
127
+ ### Results
128
+
129
+ [More Information Needed]
130
+
131
+ #### Summary
132
+
133
+
134
+
135
+ ## Model Examination [optional]
136
+
137
+ <!-- Relevant interpretability work for the model goes here -->
138
+
139
+ [More Information Needed]
140
+
141
+ ## Environmental Impact
142
+
143
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
144
+
145
+ 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).
146
+
147
+ - **Hardware Type:** [More Information Needed]
148
+ - **Hours used:** [More Information Needed]
149
+ - **Cloud Provider:** [More Information Needed]
150
+ - **Compute Region:** [More Information Needed]
151
+ - **Carbon Emitted:** [More Information Needed]
152
+
153
+ ## Technical Specifications [optional]
154
+
155
+ ### Model Architecture and Objective
156
+
157
+ [More Information Needed]
158
+
159
+ ### Compute Infrastructure
160
+
161
+ [More Information Needed]
162
+
163
+ #### Hardware
164
+
165
+ [More Information Needed]
166
+
167
+ #### Software
168
+
169
+ [More Information Needed]
170
+
171
+ ## Citation [optional]
172
+
173
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
174
+
175
+ **BibTeX:**
176
+
177
+ [More Information Needed]
178
+
179
+ **APA:**
180
+
181
+ [More Information Needed]
182
+
183
+ ## Glossary [optional]
184
+
185
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
186
+
187
+ [More Information Needed]
188
+
189
+ ## More Information [optional]
190
+
191
+ [More Information Needed]
192
+
193
+ ## Model Card Authors [optional]
194
+
195
+ [More Information Needed]
196
+
197
+ ## Model Card Contact
198
+
199
+ [More Information Needed]
200
+
201
+
202
+ ## Training procedure
203
+
204
+
205
+ The following `bitsandbytes` quantization config was used during training:
206
+ - quant_method: bitsandbytes
207
+ - load_in_8bit: False
208
+ - load_in_4bit: True
209
+ - llm_int8_threshold: 6.0
210
+ - llm_int8_skip_modules: None
211
+ - llm_int8_enable_fp32_cpu_offload: False
212
+ - llm_int8_has_fp16_weight: False
213
+ - bnb_4bit_quant_type: nf4
214
+ - bnb_4bit_use_double_quant: True
215
+ - bnb_4bit_compute_dtype: bfloat16
216
+
217
+ ### Framework versions
218
+
219
+
220
+ - PEFT 0.6.0
checkpoint-89/adapter_config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "mistralai/Mistral-7B-v0.1",
5
+ "bias": "none",
6
+ "fan_in_fan_out": null,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layers_pattern": null,
10
+ "layers_to_transform": null,
11
+ "lora_alpha": 16,
12
+ "lora_dropout": 0.05,
13
+ "modules_to_save": null,
14
+ "peft_type": "LORA",
15
+ "r": 32,
16
+ "rank_pattern": {},
17
+ "revision": null,
18
+ "target_modules": [
19
+ "o_proj",
20
+ "down_proj",
21
+ "k_proj",
22
+ "up_proj",
23
+ "v_proj",
24
+ "gate_proj",
25
+ "q_proj"
26
+ ],
27
+ "task_type": "CAUSAL_LM"
28
+ }
checkpoint-89/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0dabb64b2b9676fc7cf877e9237b6ce7d4e33dc0d31e3b43f65d66e7fe94add4
3
+ size 335604696
checkpoint-89/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a45e0b0e670a6b3e62346ebd7c56a546b0c1841277a345e413fc624d79c09506
3
+ size 168624287
checkpoint-89/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a73475f16f9a6f1784d28cd377e784fc0fc969c595e3e0ce6b3b0a05dcfc365
3
+ size 14575
checkpoint-89/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48689e5a148444ef4ecee141f4ca0896dd03e488770e55606487d4085fc01001
3
+ size 627
checkpoint-89/trainer_state.json ADDED
@@ -0,0 +1,697 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 0.9916434540389972,
5
+ "eval_steps": 5,
6
+ "global_step": 89,
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.01,
13
+ "learning_rate": 2e-05,
14
+ "loss": 1.5123,
15
+ "step": 1
16
+ },
17
+ {
18
+ "epoch": 0.01,
19
+ "eval_loss": 1.5038223266601562,
20
+ "eval_runtime": 44.7062,
21
+ "eval_samples_per_second": 6.822,
22
+ "eval_steps_per_second": 3.422,
23
+ "step": 1
24
+ },
25
+ {
26
+ "epoch": 0.02,
27
+ "learning_rate": 4e-05,
28
+ "loss": 1.491,
29
+ "step": 2
30
+ },
31
+ {
32
+ "epoch": 0.03,
33
+ "learning_rate": 6e-05,
34
+ "loss": 1.4859,
35
+ "step": 3
36
+ },
37
+ {
38
+ "epoch": 0.04,
39
+ "learning_rate": 8e-05,
40
+ "loss": 1.4147,
41
+ "step": 4
42
+ },
43
+ {
44
+ "epoch": 0.06,
45
+ "learning_rate": 0.0001,
46
+ "loss": 1.3662,
47
+ "step": 5
48
+ },
49
+ {
50
+ "epoch": 0.06,
51
+ "eval_loss": 1.4103038311004639,
52
+ "eval_runtime": 44.789,
53
+ "eval_samples_per_second": 6.81,
54
+ "eval_steps_per_second": 3.416,
55
+ "step": 5
56
+ },
57
+ {
58
+ "epoch": 0.07,
59
+ "learning_rate": 0.00012,
60
+ "loss": 1.3399,
61
+ "step": 6
62
+ },
63
+ {
64
+ "epoch": 0.08,
65
+ "learning_rate": 0.00014,
66
+ "loss": 1.346,
67
+ "step": 7
68
+ },
69
+ {
70
+ "epoch": 0.09,
71
+ "learning_rate": 0.00016,
72
+ "loss": 1.3327,
73
+ "step": 8
74
+ },
75
+ {
76
+ "epoch": 0.1,
77
+ "learning_rate": 0.00018,
78
+ "loss": 1.3466,
79
+ "step": 9
80
+ },
81
+ {
82
+ "epoch": 0.11,
83
+ "learning_rate": 0.0002,
84
+ "loss": 1.1836,
85
+ "step": 10
86
+ },
87
+ {
88
+ "epoch": 0.11,
89
+ "eval_loss": 1.3055061101913452,
90
+ "eval_runtime": 44.7824,
91
+ "eval_samples_per_second": 6.811,
92
+ "eval_steps_per_second": 3.417,
93
+ "step": 10
94
+ },
95
+ {
96
+ "epoch": 0.12,
97
+ "learning_rate": 0.00019992093972273018,
98
+ "loss": 1.3094,
99
+ "step": 11
100
+ },
101
+ {
102
+ "epoch": 0.13,
103
+ "learning_rate": 0.0001996838839014696,
104
+ "loss": 1.3395,
105
+ "step": 12
106
+ },
107
+ {
108
+ "epoch": 0.14,
109
+ "learning_rate": 0.00019928920737019733,
110
+ "loss": 1.2689,
111
+ "step": 13
112
+ },
113
+ {
114
+ "epoch": 0.16,
115
+ "learning_rate": 0.00019873753419363336,
116
+ "loss": 1.2584,
117
+ "step": 14
118
+ },
119
+ {
120
+ "epoch": 0.17,
121
+ "learning_rate": 0.00019802973668046363,
122
+ "loss": 1.2761,
123
+ "step": 15
124
+ },
125
+ {
126
+ "epoch": 0.17,
127
+ "eval_loss": 1.2809727191925049,
128
+ "eval_runtime": 44.7653,
129
+ "eval_samples_per_second": 6.813,
130
+ "eval_steps_per_second": 3.418,
131
+ "step": 15
132
+ },
133
+ {
134
+ "epoch": 0.18,
135
+ "learning_rate": 0.000197166934004041,
136
+ "loss": 1.2508,
137
+ "step": 16
138
+ },
139
+ {
140
+ "epoch": 0.19,
141
+ "learning_rate": 0.00019615049043274205,
142
+ "loss": 1.151,
143
+ "step": 17
144
+ },
145
+ {
146
+ "epoch": 0.2,
147
+ "learning_rate": 0.00019498201317277828,
148
+ "loss": 1.2186,
149
+ "step": 18
150
+ },
151
+ {
152
+ "epoch": 0.21,
153
+ "learning_rate": 0.0001936633498268728,
154
+ "loss": 1.2416,
155
+ "step": 19
156
+ },
157
+ {
158
+ "epoch": 0.22,
159
+ "learning_rate": 0.00019219658547282067,
160
+ "loss": 1.1779,
161
+ "step": 20
162
+ },
163
+ {
164
+ "epoch": 0.22,
165
+ "eval_loss": 1.2696019411087036,
166
+ "eval_runtime": 44.7761,
167
+ "eval_samples_per_second": 6.812,
168
+ "eval_steps_per_second": 3.417,
169
+ "step": 20
170
+ },
171
+ {
172
+ "epoch": 0.23,
173
+ "learning_rate": 0.00019058403936655233,
174
+ "loss": 1.2289,
175
+ "step": 21
176
+ },
177
+ {
178
+ "epoch": 0.25,
179
+ "learning_rate": 0.0001888282612749132,
180
+ "loss": 1.2383,
181
+ "step": 22
182
+ },
183
+ {
184
+ "epoch": 0.26,
185
+ "learning_rate": 0.00018693202744395827,
186
+ "loss": 1.2303,
187
+ "step": 23
188
+ },
189
+ {
190
+ "epoch": 0.27,
191
+ "learning_rate": 0.00018489833620913642,
192
+ "loss": 1.0146,
193
+ "step": 24
194
+ },
195
+ {
196
+ "epoch": 0.28,
197
+ "learning_rate": 0.00018273040325430574,
198
+ "loss": 1.1242,
199
+ "step": 25
200
+ },
201
+ {
202
+ "epoch": 0.28,
203
+ "eval_loss": 1.264197587966919,
204
+ "eval_runtime": 44.7709,
205
+ "eval_samples_per_second": 6.812,
206
+ "eval_steps_per_second": 3.417,
207
+ "step": 25
208
+ },
209
+ {
210
+ "epoch": 0.29,
211
+ "learning_rate": 0.00018043165652707649,
212
+ "loss": 1.1922,
213
+ "step": 26
214
+ },
215
+ {
216
+ "epoch": 0.3,
217
+ "learning_rate": 0.00017800573081852122,
218
+ "loss": 1.2137,
219
+ "step": 27
220
+ },
221
+ {
222
+ "epoch": 0.31,
223
+ "learning_rate": 0.00017545646201582303,
224
+ "loss": 1.2167,
225
+ "step": 28
226
+ },
227
+ {
228
+ "epoch": 0.32,
229
+ "learning_rate": 0.00017278788103694943,
230
+ "loss": 1.2603,
231
+ "step": 29
232
+ },
233
+ {
234
+ "epoch": 0.33,
235
+ "learning_rate": 0.00017000420745694254,
236
+ "loss": 1.2414,
237
+ "step": 30
238
+ },
239
+ {
240
+ "epoch": 0.33,
241
+ "eval_loss": 1.2588073015213013,
242
+ "eval_runtime": 44.7874,
243
+ "eval_samples_per_second": 6.81,
244
+ "eval_steps_per_second": 3.416,
245
+ "step": 30
246
+ },
247
+ {
248
+ "epoch": 0.35,
249
+ "learning_rate": 0.0001671098428359037,
250
+ "loss": 1.187,
251
+ "step": 31
252
+ },
253
+ {
254
+ "epoch": 0.36,
255
+ "learning_rate": 0.000164109363759222,
256
+ "loss": 1.165,
257
+ "step": 32
258
+ },
259
+ {
260
+ "epoch": 0.37,
261
+ "learning_rate": 0.00016100751460105243,
262
+ "loss": 1.1853,
263
+ "step": 33
264
+ },
265
+ {
266
+ "epoch": 0.38,
267
+ "learning_rate": 0.00015780920002248484,
268
+ "loss": 1.0538,
269
+ "step": 34
270
+ },
271
+ {
272
+ "epoch": 0.39,
273
+ "learning_rate": 0.00015451947721626676,
274
+ "loss": 1.1382,
275
+ "step": 35
276
+ },
277
+ {
278
+ "epoch": 0.39,
279
+ "eval_loss": 1.2555344104766846,
280
+ "eval_runtime": 44.7839,
281
+ "eval_samples_per_second": 6.81,
282
+ "eval_steps_per_second": 3.416,
283
+ "step": 35
284
+ },
285
+ {
286
+ "epoch": 0.4,
287
+ "learning_rate": 0.00015114354791034225,
288
+ "loss": 1.1604,
289
+ "step": 36
290
+ },
291
+ {
292
+ "epoch": 0.41,
293
+ "learning_rate": 0.00014768675014285062,
294
+ "loss": 1.1631,
295
+ "step": 37
296
+ },
297
+ {
298
+ "epoch": 0.42,
299
+ "learning_rate": 0.0001441545498215912,
300
+ "loss": 1.2075,
301
+ "step": 38
302
+ },
303
+ {
304
+ "epoch": 0.43,
305
+ "learning_rate": 0.00014055253208129938,
306
+ "loss": 1.1602,
307
+ "step": 39
308
+ },
309
+ {
310
+ "epoch": 0.45,
311
+ "learning_rate": 0.00013688639245240078,
312
+ "loss": 1.2094,
313
+ "step": 40
314
+ },
315
+ {
316
+ "epoch": 0.45,
317
+ "eval_loss": 1.2520103454589844,
318
+ "eval_runtime": 44.7798,
319
+ "eval_samples_per_second": 6.811,
320
+ "eval_steps_per_second": 3.417,
321
+ "step": 40
322
+ },
323
+ {
324
+ "epoch": 0.46,
325
+ "learning_rate": 0.0001331619278552068,
326
+ "loss": 1.1457,
327
+ "step": 41
328
+ },
329
+ {
330
+ "epoch": 0.47,
331
+ "learning_rate": 0.00012938502743379212,
332
+ "loss": 1.0895,
333
+ "step": 42
334
+ },
335
+ {
336
+ "epoch": 0.48,
337
+ "learning_rate": 0.0001255616632440475,
338
+ "loss": 1.0843,
339
+ "step": 43
340
+ },
341
+ {
342
+ "epoch": 0.49,
343
+ "learning_rate": 0.0001216978808106318,
344
+ "loss": 1.1428,
345
+ "step": 44
346
+ },
347
+ {
348
+ "epoch": 0.5,
349
+ "learning_rate": 0.00011779978956775506,
350
+ "loss": 1.1049,
351
+ "step": 45
352
+ },
353
+ {
354
+ "epoch": 0.5,
355
+ "eval_loss": 1.2504087686538696,
356
+ "eval_runtime": 44.7786,
357
+ "eval_samples_per_second": 6.811,
358
+ "eval_steps_per_second": 3.417,
359
+ "step": 45
360
+ },
361
+ {
362
+ "epoch": 0.51,
363
+ "learning_rate": 0.00011387355319890685,
364
+ "loss": 1.0035,
365
+ "step": 46
366
+ },
367
+ {
368
+ "epoch": 0.52,
369
+ "learning_rate": 0.00010992537989080618,
370
+ "loss": 1.1693,
371
+ "step": 47
372
+ },
373
+ {
374
+ "epoch": 0.53,
375
+ "learning_rate": 0.00010596151251698199,
376
+ "loss": 1.0852,
377
+ "step": 48
378
+ },
379
+ {
380
+ "epoch": 0.55,
381
+ "learning_rate": 0.00010198821876650701,
382
+ "loss": 1.0954,
383
+ "step": 49
384
+ },
385
+ {
386
+ "epoch": 0.56,
387
+ "learning_rate": 9.801178123349298e-05,
388
+ "loss": 1.1709,
389
+ "step": 50
390
+ },
391
+ {
392
+ "epoch": 0.56,
393
+ "eval_loss": 1.2487094402313232,
394
+ "eval_runtime": 44.7848,
395
+ "eval_samples_per_second": 6.81,
396
+ "eval_steps_per_second": 3.416,
397
+ "step": 50
398
+ },
399
+ {
400
+ "epoch": 0.57,
401
+ "learning_rate": 9.403848748301802e-05,
402
+ "loss": 1.1668,
403
+ "step": 51
404
+ },
405
+ {
406
+ "epoch": 0.58,
407
+ "learning_rate": 9.007462010919386e-05,
408
+ "loss": 1.1643,
409
+ "step": 52
410
+ },
411
+ {
412
+ "epoch": 0.59,
413
+ "learning_rate": 8.612644680109319e-05,
414
+ "loss": 1.1938,
415
+ "step": 53
416
+ },
417
+ {
418
+ "epoch": 0.6,
419
+ "learning_rate": 8.2200210432245e-05,
420
+ "loss": 1.1062,
421
+ "step": 54
422
+ },
423
+ {
424
+ "epoch": 0.61,
425
+ "learning_rate": 7.83021191893682e-05,
426
+ "loss": 1.0981,
427
+ "step": 55
428
+ },
429
+ {
430
+ "epoch": 0.61,
431
+ "eval_loss": 1.2463428974151611,
432
+ "eval_runtime": 44.7606,
433
+ "eval_samples_per_second": 6.814,
434
+ "eval_steps_per_second": 3.418,
435
+ "step": 55
436
+ },
437
+ {
438
+ "epoch": 0.62,
439
+ "learning_rate": 7.443833675595255e-05,
440
+ "loss": 1.1809,
441
+ "step": 56
442
+ },
443
+ {
444
+ "epoch": 0.64,
445
+ "learning_rate": 7.061497256620793e-05,
446
+ "loss": 1.1515,
447
+ "step": 57
448
+ },
449
+ {
450
+ "epoch": 0.65,
451
+ "learning_rate": 6.683807214479323e-05,
452
+ "loss": 1.1226,
453
+ "step": 58
454
+ },
455
+ {
456
+ "epoch": 0.66,
457
+ "learning_rate": 6.311360754759923e-05,
458
+ "loss": 1.1952,
459
+ "step": 59
460
+ },
461
+ {
462
+ "epoch": 0.67,
463
+ "learning_rate": 5.9447467918700614e-05,
464
+ "loss": 1.1902,
465
+ "step": 60
466
+ },
467
+ {
468
+ "epoch": 0.67,
469
+ "eval_loss": 1.2446388006210327,
470
+ "eval_runtime": 44.7804,
471
+ "eval_samples_per_second": 6.811,
472
+ "eval_steps_per_second": 3.417,
473
+ "step": 60
474
+ },
475
+ {
476
+ "epoch": 0.68,
477
+ "learning_rate": 5.584545017840885e-05,
478
+ "loss": 1.1506,
479
+ "step": 61
480
+ },
481
+ {
482
+ "epoch": 0.69,
483
+ "learning_rate": 5.2313249857149414e-05,
484
+ "loss": 1.1505,
485
+ "step": 62
486
+ },
487
+ {
488
+ "epoch": 0.7,
489
+ "learning_rate": 4.885645208965779e-05,
490
+ "loss": 1.1414,
491
+ "step": 63
492
+ },
493
+ {
494
+ "epoch": 0.71,
495
+ "learning_rate": 4.548052278373327e-05,
496
+ "loss": 1.2361,
497
+ "step": 64
498
+ },
499
+ {
500
+ "epoch": 0.72,
501
+ "learning_rate": 4.219079997751515e-05,
502
+ "loss": 1.1526,
503
+ "step": 65
504
+ },
505
+ {
506
+ "epoch": 0.72,
507
+ "eval_loss": 1.244616150856018,
508
+ "eval_runtime": 44.7366,
509
+ "eval_samples_per_second": 6.818,
510
+ "eval_steps_per_second": 3.42,
511
+ "step": 65
512
+ },
513
+ {
514
+ "epoch": 0.74,
515
+ "learning_rate": 3.899248539894757e-05,
516
+ "loss": 1.1942,
517
+ "step": 66
518
+ },
519
+ {
520
+ "epoch": 0.75,
521
+ "learning_rate": 3.589063624077802e-05,
522
+ "loss": 1.1018,
523
+ "step": 67
524
+ },
525
+ {
526
+ "epoch": 0.76,
527
+ "learning_rate": 3.289015716409631e-05,
528
+ "loss": 1.2284,
529
+ "step": 68
530
+ },
531
+ {
532
+ "epoch": 0.77,
533
+ "learning_rate": 2.9995792543057478e-05,
534
+ "loss": 1.1665,
535
+ "step": 69
536
+ },
537
+ {
538
+ "epoch": 0.78,
539
+ "learning_rate": 2.7212118963050592e-05,
540
+ "loss": 1.1319,
541
+ "step": 70
542
+ },
543
+ {
544
+ "epoch": 0.78,
545
+ "eval_loss": 1.2439743280410767,
546
+ "eval_runtime": 44.7492,
547
+ "eval_samples_per_second": 6.816,
548
+ "eval_steps_per_second": 3.419,
549
+ "step": 70
550
+ },
551
+ {
552
+ "epoch": 0.79,
553
+ "learning_rate": 2.4543537984176978e-05,
554
+ "loss": 1.067,
555
+ "step": 71
556
+ },
557
+ {
558
+ "epoch": 0.8,
559
+ "learning_rate": 2.19942691814788e-05,
560
+ "loss": 1.1465,
561
+ "step": 72
562
+ },
563
+ {
564
+ "epoch": 0.81,
565
+ "learning_rate": 1.9568343472923524e-05,
566
+ "loss": 1.227,
567
+ "step": 73
568
+ },
569
+ {
570
+ "epoch": 0.82,
571
+ "learning_rate": 1.7269596745694295e-05,
572
+ "loss": 1.1989,
573
+ "step": 74
574
+ },
575
+ {
576
+ "epoch": 0.84,
577
+ "learning_rate": 1.5101663790863596e-05,
578
+ "loss": 1.1913,
579
+ "step": 75
580
+ },
581
+ {
582
+ "epoch": 0.84,
583
+ "eval_loss": 1.2430416345596313,
584
+ "eval_runtime": 44.7589,
585
+ "eval_samples_per_second": 6.814,
586
+ "eval_steps_per_second": 3.418,
587
+ "step": 75
588
+ },
589
+ {
590
+ "epoch": 0.85,
591
+ "learning_rate": 1.3067972556041752e-05,
592
+ "loss": 1.154,
593
+ "step": 76
594
+ },
595
+ {
596
+ "epoch": 0.86,
597
+ "learning_rate": 1.1171738725086833e-05,
598
+ "loss": 1.055,
599
+ "step": 77
600
+ },
601
+ {
602
+ "epoch": 0.87,
603
+ "learning_rate": 9.415960633447674e-06,
604
+ "loss": 1.1523,
605
+ "step": 78
606
+ },
607
+ {
608
+ "epoch": 0.88,
609
+ "learning_rate": 7.803414527179343e-06,
610
+ "loss": 1.1057,
611
+ "step": 79
612
+ },
613
+ {
614
+ "epoch": 0.89,
615
+ "learning_rate": 6.336650173127223e-06,
616
+ "loss": 1.1875,
617
+ "step": 80
618
+ },
619
+ {
620
+ "epoch": 0.89,
621
+ "eval_loss": 1.2423635721206665,
622
+ "eval_runtime": 44.7706,
623
+ "eval_samples_per_second": 6.813,
624
+ "eval_steps_per_second": 3.417,
625
+ "step": 80
626
+ },
627
+ {
628
+ "epoch": 0.9,
629
+ "learning_rate": 5.017986827221733e-06,
630
+ "loss": 1.2235,
631
+ "step": 81
632
+ },
633
+ {
634
+ "epoch": 0.91,
635
+ "learning_rate": 3.849509567257959e-06,
636
+ "loss": 1.1457,
637
+ "step": 82
638
+ },
639
+ {
640
+ "epoch": 0.92,
641
+ "learning_rate": 2.8330659959589946e-06,
642
+ "loss": 1.1552,
643
+ "step": 83
644
+ },
645
+ {
646
+ "epoch": 0.94,
647
+ "learning_rate": 1.9702633195363917e-06,
648
+ "loss": 1.1497,
649
+ "step": 84
650
+ },
651
+ {
652
+ "epoch": 0.95,
653
+ "learning_rate": 1.2624658063666639e-06,
654
+ "loss": 1.1454,
655
+ "step": 85
656
+ },
657
+ {
658
+ "epoch": 0.95,
659
+ "eval_loss": 1.2422702312469482,
660
+ "eval_runtime": 44.7707,
661
+ "eval_samples_per_second": 6.812,
662
+ "eval_steps_per_second": 3.417,
663
+ "step": 85
664
+ },
665
+ {
666
+ "epoch": 0.96,
667
+ "learning_rate": 7.10792629802659e-07,
668
+ "loss": 1.2008,
669
+ "step": 86
670
+ },
671
+ {
672
+ "epoch": 0.97,
673
+ "learning_rate": 3.161160985304168e-07,
674
+ "loss": 1.15,
675
+ "step": 87
676
+ },
677
+ {
678
+ "epoch": 0.98,
679
+ "learning_rate": 7.906027726981568e-08,
680
+ "loss": 1.2256,
681
+ "step": 88
682
+ },
683
+ {
684
+ "epoch": 0.99,
685
+ "learning_rate": 0.0,
686
+ "loss": 1.2861,
687
+ "step": 89
688
+ }
689
+ ],
690
+ "logging_steps": 1,
691
+ "max_steps": 89,
692
+ "num_train_epochs": 1,
693
+ "save_steps": 500,
694
+ "total_flos": 2.5178194955363942e+17,
695
+ "trial_name": null,
696
+ "trial_params": null
697
+ }
checkpoint-89/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8bdfe72d52e79a8c86654ed5bfb76054ce589a732b55e1e4b5cdc391ffcc1469
3
+ size 4539
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "mistralai/Mistral-7B-v0.1",
3
+ "architectures": [
4
+ "MistralForCausalLM"
5
+ ],
6
+ "bos_token_id": 1,
7
+ "eos_token_id": 2,
8
+ "hidden_act": "silu",
9
+ "hidden_size": 4096,
10
+ "initializer_range": 0.02,
11
+ "intermediate_size": 14336,
12
+ "max_position_embeddings": 32768,
13
+ "model_type": "mistral",
14
+ "num_attention_heads": 32,
15
+ "num_hidden_layers": 32,
16
+ "num_key_value_heads": 8,
17
+ "quantization_config": {
18
+ "bnb_4bit_compute_dtype": "bfloat16",
19
+ "bnb_4bit_quant_type": "nf4",
20
+ "bnb_4bit_use_double_quant": true,
21
+ "llm_int8_enable_fp32_cpu_offload": false,
22
+ "llm_int8_has_fp16_weight": false,
23
+ "llm_int8_skip_modules": null,
24
+ "llm_int8_threshold": 6.0,
25
+ "load_in_4bit": true,
26
+ "load_in_8bit": false,
27
+ "quant_method": "bitsandbytes"
28
+ },
29
+ "rms_norm_eps": 1e-05,
30
+ "rope_theta": 10000.0,
31
+ "sliding_window": 4096,
32
+ "tie_word_embeddings": false,
33
+ "torch_dtype": "bfloat16",
34
+ "transformers_version": "4.35.2",
35
+ "use_cache": false,
36
+ "vocab_size": 32000
37
+ }
runs/Dec06_16-57-37_f03522d91cb7/events.out.tfevents.1701881857.f03522d91cb7.675.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e671306047a1ad8ff5a26ce57424292eeab70c2a741db3599c7beb1ecc5e384
3
+ size 23848
special_tokens_map.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": "</s>",
17
+ "unk_token": {
18
+ "content": "<unk>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ }
24
+ }
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dadfd56d766715c61d2ef780a525ab43b8e6da4de6865bda3d95fdef5e134055
3
+ size 493443
tokenizer_config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "added_tokens_decoder": {
5
+ "0": {
6
+ "content": "<unk>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false,
11
+ "special": true
12
+ },
13
+ "1": {
14
+ "content": "<s>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false,
19
+ "special": true
20
+ },
21
+ "2": {
22
+ "content": "</s>",
23
+ "lstrip": false,
24
+ "normalized": false,
25
+ "rstrip": false,
26
+ "single_word": false,
27
+ "special": true
28
+ }
29
+ },
30
+ "additional_special_tokens": [],
31
+ "bos_token": "<s>",
32
+ "clean_up_tokenization_spaces": false,
33
+ "eos_token": "</s>",
34
+ "legacy": true,
35
+ "model_max_length": 1000000000000000019884624838656,
36
+ "pad_token": "</s>",
37
+ "sp_model_kwargs": {},
38
+ "spaces_between_special_tokens": false,
39
+ "tokenizer_class": "LlamaTokenizer",
40
+ "trust_remote_code": false,
41
+ "unk_token": "<unk>",
42
+ "use_default_system_prompt": true,
43
+ "use_fast": true
44
+ }