jasonmsetiadi commited on
Commit
fdb739b
·
verified ·
1 Parent(s): 9eef2f2

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,142 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: peft
4
+ tags:
5
+ - generated_from_trainer
6
+ base_model: TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
7
+ model-index:
8
+ - name: outputs/qlora-out
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ [<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)
16
+ <details><summary>See axolotl config</summary>
17
+
18
+ axolotl version: `0.4.1`
19
+ ```yaml
20
+ base_model: TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
21
+ model_type: LlamaForCausalLM
22
+ tokenizer_type: LlamaTokenizer
23
+
24
+ load_in_8bit: false
25
+ load_in_4bit: true
26
+ strict: false
27
+
28
+ datasets:
29
+ - path: mhenrichsen/alpaca_2k_test
30
+ type: alpaca
31
+ dataset_prepared_path:
32
+ val_set_size: 0.05
33
+ output_dir: ./outputs/qlora-out
34
+
35
+ adapter: qlora
36
+ lora_model_dir:
37
+
38
+ sequence_len: 4096
39
+ sample_packing: true
40
+ eval_sample_packing: false
41
+ pad_to_sequence_len: true
42
+
43
+ lora_r: 32
44
+ lora_alpha: 16
45
+ lora_dropout: 0.05
46
+ lora_target_modules:
47
+ lora_target_linear: true
48
+ lora_fan_in_fan_out:
49
+
50
+ wandb_project:
51
+ wandb_entity:
52
+ wandb_watch:
53
+ wandb_name:
54
+ wandb_log_model:
55
+
56
+ gradient_accumulation_steps: 4
57
+ micro_batch_size: 1
58
+ num_epochs: 1
59
+ optimizer: paged_adamw_32bit
60
+ lr_scheduler: cosine
61
+ learning_rate: 0.0002
62
+
63
+ train_on_inputs: false
64
+ group_by_length: false
65
+ bf16: auto
66
+ fp16:
67
+ tf32: false
68
+
69
+ gradient_checkpointing: true
70
+ early_stopping_patience:
71
+ resume_from_checkpoint:
72
+ local_rank:
73
+ logging_steps: 1
74
+ xformers_attention:
75
+ flash_attention: true
76
+
77
+ warmup_steps: 10
78
+ evals_per_epoch: 4
79
+ saves_per_epoch: 1
80
+ debug:
81
+ deepspeed:
82
+ weight_decay: 0.0
83
+ fsdp:
84
+ fsdp_config:
85
+ special_tokens:
86
+
87
+ ```
88
+
89
+ </details><br>
90
+
91
+ # outputs/qlora-out
92
+
93
+ This model is a fine-tuned version of [TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T](https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T) on the None dataset.
94
+ It achieves the following results on the evaluation set:
95
+ - Loss: 1.2355
96
+
97
+ ## Model description
98
+
99
+ More information needed
100
+
101
+ ## Intended uses & limitations
102
+
103
+ More information needed
104
+
105
+ ## Training and evaluation data
106
+
107
+ More information needed
108
+
109
+ ## Training procedure
110
+
111
+ ### Training hyperparameters
112
+
113
+ The following hyperparameters were used during training:
114
+ - learning_rate: 0.0002
115
+ - train_batch_size: 1
116
+ - eval_batch_size: 1
117
+ - seed: 42
118
+ - gradient_accumulation_steps: 4
119
+ - total_train_batch_size: 4
120
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
121
+ - lr_scheduler_type: cosine
122
+ - lr_scheduler_warmup_steps: 10
123
+ - num_epochs: 1
124
+
125
+ ### Training results
126
+
127
+ | Training Loss | Epoch | Step | Validation Loss |
128
+ |:-------------:|:------:|:----:|:---------------:|
129
+ | 1.5483 | 0.0404 | 1 | 1.5657 |
130
+ | 1.3687 | 0.2424 | 6 | 1.4985 |
131
+ | 1.1948 | 0.4848 | 12 | 1.2933 |
132
+ | 1.2227 | 0.7273 | 18 | 1.2485 |
133
+ | 1.2483 | 0.9697 | 24 | 1.2355 |
134
+
135
+
136
+ ### Framework versions
137
+
138
+ - PEFT 0.11.1
139
+ - Transformers 4.41.1
140
+ - Pytorch 2.1.2+cu118
141
+ - Datasets 2.19.1
142
+ - Tokenizers 0.19.1
adapter_config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T",
5
+ "bias": "none",
6
+ "fan_in_fan_out": null,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layer_replication": null,
10
+ "layers_pattern": null,
11
+ "layers_to_transform": null,
12
+ "loftq_config": {},
13
+ "lora_alpha": 16,
14
+ "lora_dropout": 0.05,
15
+ "megatron_config": null,
16
+ "megatron_core": "megatron.core",
17
+ "modules_to_save": null,
18
+ "peft_type": "LORA",
19
+ "r": 32,
20
+ "rank_pattern": {},
21
+ "revision": null,
22
+ "target_modules": [
23
+ "q_proj",
24
+ "up_proj",
25
+ "k_proj",
26
+ "down_proj",
27
+ "gate_proj",
28
+ "o_proj",
29
+ "v_proj"
30
+ ],
31
+ "task_type": "CAUSAL_LM",
32
+ "use_dora": false,
33
+ "use_rslora": false
34
+ }
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:28f686208eb603e28e5e3529175b50c0d07484414fec46c37883fa2f2f6cf363
3
+ size 50573978
checkpoint-24/README.md ADDED
@@ -0,0 +1,202 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ base_model: TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
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
+ ### Framework versions
201
+
202
+ - PEFT 0.11.1
checkpoint-24/adapter_config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T",
5
+ "bias": "none",
6
+ "fan_in_fan_out": null,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layer_replication": null,
10
+ "layers_pattern": null,
11
+ "layers_to_transform": null,
12
+ "loftq_config": {},
13
+ "lora_alpha": 16,
14
+ "lora_dropout": 0.05,
15
+ "megatron_config": null,
16
+ "megatron_core": "megatron.core",
17
+ "modules_to_save": null,
18
+ "peft_type": "LORA",
19
+ "r": 32,
20
+ "rank_pattern": {},
21
+ "revision": null,
22
+ "target_modules": [
23
+ "q_proj",
24
+ "up_proj",
25
+ "k_proj",
26
+ "down_proj",
27
+ "gate_proj",
28
+ "o_proj",
29
+ "v_proj"
30
+ ],
31
+ "task_type": "CAUSAL_LM",
32
+ "use_dora": false,
33
+ "use_rslora": false
34
+ }
checkpoint-24/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7547dc708818163783e2c5c5301f5a594c041ed26c8759382e7c81b9bf973900
3
+ size 50503848
checkpoint-24/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88fe0680e396c979940b2ba39db8607c06b577c6517955b3d4af7e29413fac45
3
+ size 202035450
checkpoint-24/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe9383702bcc6b042db1e0a90b6914d41b42be7ecc713b3a77a89c8ee7e9f339
3
+ size 14244
checkpoint-24/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe253fd3c50869b3b413fa44d3f808dd4b8eee9f82cff1555f3e301210ae2483
3
+ size 1064
checkpoint-24/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
+ }
checkpoint-24/tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e556afd44213b6bd1be2b850ebbbd98f5481437a8021afaf58ee7fb1818d347
3
+ size 499723
checkpoint-24/tokenizer_config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": true,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<unk>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<s>",
16
+ "lstrip": false,
17
+ "normalized": false,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "2": {
23
+ "content": "</s>",
24
+ "lstrip": false,
25
+ "normalized": false,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": true
29
+ }
30
+ },
31
+ "bos_token": "<s>",
32
+ "clean_up_tokenization_spaces": false,
33
+ "eos_token": "</s>",
34
+ "legacy": false,
35
+ "model_max_length": 1000000000000000019884624838656,
36
+ "pad_token": "</s>",
37
+ "padding_side": "right",
38
+ "sp_model_kwargs": {},
39
+ "spaces_between_special_tokens": false,
40
+ "tokenizer_class": "LlamaTokenizer",
41
+ "unk_token": "<unk>",
42
+ "use_default_system_prompt": false,
43
+ "use_fast": true
44
+ }
checkpoint-24/trainer_state.json ADDED
@@ -0,0 +1,241 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 0.9696969696969697,
5
+ "eval_steps": 6,
6
+ "global_step": 24,
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.04040404040404041,
13
+ "grad_norm": 0.1962890625,
14
+ "learning_rate": 2e-05,
15
+ "loss": 1.5483,
16
+ "step": 1
17
+ },
18
+ {
19
+ "epoch": 0.04040404040404041,
20
+ "eval_loss": 1.565699815750122,
21
+ "eval_runtime": 10.0556,
22
+ "eval_samples_per_second": 9.945,
23
+ "eval_steps_per_second": 9.945,
24
+ "step": 1
25
+ },
26
+ {
27
+ "epoch": 0.08080808080808081,
28
+ "grad_norm": 0.1943359375,
29
+ "learning_rate": 4e-05,
30
+ "loss": 1.4502,
31
+ "step": 2
32
+ },
33
+ {
34
+ "epoch": 0.12121212121212122,
35
+ "grad_norm": 0.1962890625,
36
+ "learning_rate": 6e-05,
37
+ "loss": 1.416,
38
+ "step": 3
39
+ },
40
+ {
41
+ "epoch": 0.16161616161616163,
42
+ "grad_norm": 0.24609375,
43
+ "learning_rate": 8e-05,
44
+ "loss": 1.5064,
45
+ "step": 4
46
+ },
47
+ {
48
+ "epoch": 0.20202020202020202,
49
+ "grad_norm": 0.1982421875,
50
+ "learning_rate": 0.0001,
51
+ "loss": 1.448,
52
+ "step": 5
53
+ },
54
+ {
55
+ "epoch": 0.24242424242424243,
56
+ "grad_norm": 0.1767578125,
57
+ "learning_rate": 0.00012,
58
+ "loss": 1.3687,
59
+ "step": 6
60
+ },
61
+ {
62
+ "epoch": 0.24242424242424243,
63
+ "eval_loss": 1.4984517097473145,
64
+ "eval_runtime": 10.4286,
65
+ "eval_samples_per_second": 9.589,
66
+ "eval_steps_per_second": 9.589,
67
+ "step": 6
68
+ },
69
+ {
70
+ "epoch": 0.2828282828282828,
71
+ "grad_norm": 0.1796875,
72
+ "learning_rate": 0.00014,
73
+ "loss": 1.2437,
74
+ "step": 7
75
+ },
76
+ {
77
+ "epoch": 0.32323232323232326,
78
+ "grad_norm": 0.146484375,
79
+ "learning_rate": 0.00016,
80
+ "loss": 1.2156,
81
+ "step": 8
82
+ },
83
+ {
84
+ "epoch": 0.36363636363636365,
85
+ "grad_norm": 0.1376953125,
86
+ "learning_rate": 0.00018,
87
+ "loss": 1.281,
88
+ "step": 9
89
+ },
90
+ {
91
+ "epoch": 0.40404040404040403,
92
+ "grad_norm": 0.1259765625,
93
+ "learning_rate": 0.0002,
94
+ "loss": 1.3515,
95
+ "step": 10
96
+ },
97
+ {
98
+ "epoch": 0.4444444444444444,
99
+ "grad_norm": 0.1337890625,
100
+ "learning_rate": 0.00019749279121818235,
101
+ "loss": 1.3942,
102
+ "step": 11
103
+ },
104
+ {
105
+ "epoch": 0.48484848484848486,
106
+ "grad_norm": 0.162109375,
107
+ "learning_rate": 0.0001900968867902419,
108
+ "loss": 1.1948,
109
+ "step": 12
110
+ },
111
+ {
112
+ "epoch": 0.48484848484848486,
113
+ "eval_loss": 1.2932639122009277,
114
+ "eval_runtime": 10.761,
115
+ "eval_samples_per_second": 9.293,
116
+ "eval_steps_per_second": 9.293,
117
+ "step": 12
118
+ },
119
+ {
120
+ "epoch": 0.5252525252525253,
121
+ "grad_norm": 0.171875,
122
+ "learning_rate": 0.000178183148246803,
123
+ "loss": 1.2631,
124
+ "step": 13
125
+ },
126
+ {
127
+ "epoch": 0.5656565656565656,
128
+ "grad_norm": 0.1435546875,
129
+ "learning_rate": 0.00016234898018587337,
130
+ "loss": 1.2711,
131
+ "step": 14
132
+ },
133
+ {
134
+ "epoch": 0.6060606060606061,
135
+ "grad_norm": 0.12353515625,
136
+ "learning_rate": 0.00014338837391175582,
137
+ "loss": 1.2824,
138
+ "step": 15
139
+ },
140
+ {
141
+ "epoch": 0.6464646464646465,
142
+ "grad_norm": 0.115234375,
143
+ "learning_rate": 0.00012225209339563145,
144
+ "loss": 1.2682,
145
+ "step": 16
146
+ },
147
+ {
148
+ "epoch": 0.6868686868686869,
149
+ "grad_norm": 0.10009765625,
150
+ "learning_rate": 0.0001,
151
+ "loss": 1.1965,
152
+ "step": 17
153
+ },
154
+ {
155
+ "epoch": 0.7272727272727273,
156
+ "grad_norm": 0.11328125,
157
+ "learning_rate": 7.774790660436858e-05,
158
+ "loss": 1.2227,
159
+ "step": 18
160
+ },
161
+ {
162
+ "epoch": 0.7272727272727273,
163
+ "eval_loss": 1.2484949827194214,
164
+ "eval_runtime": 10.7358,
165
+ "eval_samples_per_second": 9.315,
166
+ "eval_steps_per_second": 9.315,
167
+ "step": 18
168
+ },
169
+ {
170
+ "epoch": 0.7676767676767676,
171
+ "grad_norm": 0.11474609375,
172
+ "learning_rate": 5.6611626088244194e-05,
173
+ "loss": 1.1904,
174
+ "step": 19
175
+ },
176
+ {
177
+ "epoch": 0.8080808080808081,
178
+ "grad_norm": 0.0986328125,
179
+ "learning_rate": 3.7651019814126654e-05,
180
+ "loss": 1.2903,
181
+ "step": 20
182
+ },
183
+ {
184
+ "epoch": 0.8484848484848485,
185
+ "grad_norm": 0.1005859375,
186
+ "learning_rate": 2.181685175319702e-05,
187
+ "loss": 1.3272,
188
+ "step": 21
189
+ },
190
+ {
191
+ "epoch": 0.8888888888888888,
192
+ "grad_norm": 0.140625,
193
+ "learning_rate": 9.903113209758096e-06,
194
+ "loss": 1.2255,
195
+ "step": 22
196
+ },
197
+ {
198
+ "epoch": 0.9292929292929293,
199
+ "grad_norm": 0.0986328125,
200
+ "learning_rate": 2.5072087818176382e-06,
201
+ "loss": 1.2386,
202
+ "step": 23
203
+ },
204
+ {
205
+ "epoch": 0.9696969696969697,
206
+ "grad_norm": 0.107421875,
207
+ "learning_rate": 0.0,
208
+ "loss": 1.2483,
209
+ "step": 24
210
+ },
211
+ {
212
+ "epoch": 0.9696969696969697,
213
+ "eval_loss": 1.235521674156189,
214
+ "eval_runtime": 10.7486,
215
+ "eval_samples_per_second": 9.304,
216
+ "eval_steps_per_second": 9.304,
217
+ "step": 24
218
+ }
219
+ ],
220
+ "logging_steps": 1,
221
+ "max_steps": 24,
222
+ "num_input_tokens_seen": 0,
223
+ "num_train_epochs": 1,
224
+ "save_steps": 500,
225
+ "stateful_callbacks": {
226
+ "TrainerControl": {
227
+ "args": {
228
+ "should_epoch_stop": false,
229
+ "should_evaluate": false,
230
+ "should_log": false,
231
+ "should_save": true,
232
+ "should_training_stop": true
233
+ },
234
+ "attributes": {}
235
+ }
236
+ },
237
+ "total_flos": 2500249176244224.0,
238
+ "train_batch_size": 1,
239
+ "trial_name": null,
240
+ "trial_params": null
241
+ }
checkpoint-24/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd33c2283bc14201b2f733e1074a1b0d35d998408f6bcfc6be41b017f3de9bcc
3
+ size 6008
config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T",
3
+ "architectures": [
4
+ "LlamaForCausalLM"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "bos_token_id": 1,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "silu",
11
+ "hidden_size": 2048,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 5632,
14
+ "max_position_embeddings": 4096,
15
+ "mlp_bias": false,
16
+ "model_type": "llama",
17
+ "num_attention_heads": 32,
18
+ "num_hidden_layers": 22,
19
+ "num_key_value_heads": 4,
20
+ "pretraining_tp": 1,
21
+ "quantization_config": {
22
+ "_load_in_4bit": true,
23
+ "_load_in_8bit": false,
24
+ "bnb_4bit_compute_dtype": "bfloat16",
25
+ "bnb_4bit_quant_storage": "bfloat16",
26
+ "bnb_4bit_quant_type": "nf4",
27
+ "bnb_4bit_use_double_quant": true,
28
+ "llm_int8_enable_fp32_cpu_offload": false,
29
+ "llm_int8_has_fp16_weight": false,
30
+ "llm_int8_skip_modules": null,
31
+ "llm_int8_threshold": 6.0,
32
+ "load_in_4bit": true,
33
+ "load_in_8bit": false,
34
+ "quant_method": "bitsandbytes"
35
+ },
36
+ "rms_norm_eps": 1e-05,
37
+ "rope_scaling": null,
38
+ "rope_theta": 10000.0,
39
+ "tie_word_embeddings": false,
40
+ "torch_dtype": "float32",
41
+ "transformers_version": "4.41.1",
42
+ "use_cache": false,
43
+ "vocab_size": 32000
44
+ }
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:9e556afd44213b6bd1be2b850ebbbd98f5481437a8021afaf58ee7fb1818d347
3
+ size 499723
tokenizer_config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": true,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<unk>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<s>",
16
+ "lstrip": false,
17
+ "normalized": false,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "2": {
23
+ "content": "</s>",
24
+ "lstrip": false,
25
+ "normalized": false,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": true
29
+ }
30
+ },
31
+ "bos_token": "<s>",
32
+ "clean_up_tokenization_spaces": false,
33
+ "eos_token": "</s>",
34
+ "legacy": false,
35
+ "model_max_length": 1000000000000000019884624838656,
36
+ "pad_token": "</s>",
37
+ "padding_side": "right",
38
+ "sp_model_kwargs": {},
39
+ "spaces_between_special_tokens": false,
40
+ "tokenizer_class": "LlamaTokenizer",
41
+ "unk_token": "<unk>",
42
+ "use_default_system_prompt": false,
43
+ "use_fast": true
44
+ }