alexredna commited on
Commit
beb9cf1
1 Parent(s): 1105a07

Training in progress, step 20

Browse files
README.md ADDED
@@ -0,0 +1,86 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: peft
4
+ tags:
5
+ - trl
6
+ - sft
7
+ - generated_from_trainer
8
+ datasets:
9
+ - generator
10
+ base_model: TinyLlama/TinyLlama-1.1B-Chat-v1.0
11
+ model-index:
12
+ - name: Tukan-1.1B-Chat-v0.1
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # Tukan-1.1B-Chat-v0.1
20
+
21
+ This model is a fine-tuned version of [TinyLlama/TinyLlama-1.1B-Chat-v1.0](https://huggingface.co/TinyLlama/TinyLlama-1.1B-Chat-v1.0) on the generator dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 1.0546
24
+
25
+ ## Model description
26
+
27
+ More information needed
28
+
29
+ ## Intended uses & limitations
30
+
31
+ More information needed
32
+
33
+ ## Training and evaluation data
34
+
35
+ More information needed
36
+
37
+ ## Training procedure
38
+
39
+ ### Training hyperparameters
40
+
41
+ The following hyperparameters were used during training:
42
+ - learning_rate: 4e-05
43
+ - train_batch_size: 6
44
+ - eval_batch_size: 4
45
+ - seed: 42
46
+ - distributed_type: multi-GPU
47
+ - gradient_accumulation_steps: 36
48
+ - total_train_batch_size: 216
49
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
50
+ - lr_scheduler_type: cosine
51
+ - num_epochs: 1
52
+
53
+ ### Training results
54
+
55
+ | Training Loss | Epoch | Step | Validation Loss |
56
+ |:-------------:|:-----:|:----:|:---------------:|
57
+ | 1.1912 | 0.28 | 10 | 1.1099 |
58
+ | 1.1238 | 0.55 | 20 | 1.0655 |
59
+ | 1.1258 | 0.83 | 30 | 1.0550 |
60
+
61
+
62
+ ### Framework versions
63
+
64
+ - Transformers 4.36.2
65
+ - Pytorch 2.2.0a0+gitd925d94
66
+ - Datasets 2.14.6
67
+ - Tokenizers 0.15.0
68
+ ## Training procedure
69
+
70
+
71
+ The following `bitsandbytes` quantization config was used during training:
72
+ - quant_method: bitsandbytes
73
+ - load_in_8bit: False
74
+ - load_in_4bit: True
75
+ - llm_int8_threshold: 6.0
76
+ - llm_int8_skip_modules: None
77
+ - llm_int8_enable_fp32_cpu_offload: False
78
+ - llm_int8_has_fp16_weight: False
79
+ - bnb_4bit_quant_type: nf4
80
+ - bnb_4bit_use_double_quant: False
81
+ - bnb_4bit_compute_dtype: float16
82
+
83
+ ### Framework versions
84
+
85
+
86
+ - PEFT 0.6.1
adapter_config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "./data/Tukan-1.1B-Chat_merged",
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
+ "o_proj",
20
+ "down_proj",
21
+ "gate_proj",
22
+ "lm_head",
23
+ "k_proj",
24
+ "q_proj",
25
+ "up_proj",
26
+ "v_proj"
27
+ ],
28
+ "task_type": "CAUSAL_LM"
29
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c56144c85bc25b19ca10ca1f4480428776f64332bf7454d9887c134827154827
3
+ size 26362152
all_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_loss": 1.0545984506607056,
4
+ "eval_runtime": 9.2861,
5
+ "eval_samples": 130,
6
+ "eval_samples_per_second": 5.061,
7
+ "eval_steps_per_second": 1.292,
8
+ "train_loss": 1.1544433269235823,
9
+ "train_runtime": 5078.5964,
10
+ "train_samples": 29726,
11
+ "train_samples_per_second": 1.537,
12
+ "train_steps_per_second": 0.007
13
+ }
config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
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": 2048,
15
+ "model_type": "llama",
16
+ "num_attention_heads": 32,
17
+ "num_hidden_layers": 22,
18
+ "num_key_value_heads": 4,
19
+ "pretraining_tp": 1,
20
+ "quantization_config": {
21
+ "bnb_4bit_compute_dtype": "float16",
22
+ "bnb_4bit_quant_type": "nf4",
23
+ "bnb_4bit_use_double_quant": false,
24
+ "llm_int8_enable_fp32_cpu_offload": false,
25
+ "llm_int8_has_fp16_weight": false,
26
+ "llm_int8_skip_modules": null,
27
+ "llm_int8_threshold": 6.0,
28
+ "load_in_4bit": true,
29
+ "load_in_8bit": false,
30
+ "quant_method": "bitsandbytes"
31
+ },
32
+ "rms_norm_eps": 1e-05,
33
+ "rope_scaling": null,
34
+ "rope_theta": 10000.0,
35
+ "tie_word_embeddings": false,
36
+ "transformers_version": "4.36.2",
37
+ "use_cache": true,
38
+ "vocab_size": 32000
39
+ }
eval_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_loss": 1.0545984506607056,
4
+ "eval_runtime": 9.2861,
5
+ "eval_samples": 130,
6
+ "eval_samples_per_second": 5.061,
7
+ "eval_steps_per_second": 1.292
8
+ }
runs/Jan12_21-18-43_98f107f1aa39/events.out.tfevents.1705094369.98f107f1aa39.690151.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:994170c0508a41ba5a33c1c454f77c84946d403bed37d8b9c11bac41b3d8d894
3
+ size 4454
runs/Jan13_19-02-46_98f107f1aa39/events.out.tfevents.1705172710.98f107f1aa39.129514.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2deb67ebf2196af0e275c3db1fea6dd00bbb8a3aec437dbf96cc46fbcb318147
3
+ size 4857
runs/Jan13_19-08-57_98f107f1aa39/events.out.tfevents.1705173116.98f107f1aa39.135205.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0418aaf4f4e34c5e8a7f4d7195be546ab8734fd13518daf5b21a9c369929b03e
3
+ size 4857
runs/Jan13_19-15-57_98f107f1aa39/events.out.tfevents.1705173410.98f107f1aa39.140623.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e01f832704f0ed5f0d028b93878374f48bde7b61f67ae7019e2583d7afc535e
3
+ size 5011
runs/Jan13_19-23-25_98f107f1aa39/events.out.tfevents.1705173982.98f107f1aa39.146842.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:826fc611d1288bf9146d9d14259f2ca0fc64fa6389bdca206612579e5e4089cd
3
+ size 4857
runs/Jan13_19-27-51_98f107f1aa39/events.out.tfevents.1705174126.98f107f1aa39.150537.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:24ec04663f35abe7f8249c5873aeac32e488d2fb900c9186f40492bf9a269f4a
3
+ size 4857
runs/Jan13_19-31-59_98f107f1aa39/events.out.tfevents.1705174497.98f107f1aa39.154347.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:19b23ffb1193103c8874ec5d6b80f396589fdaf1ea2fab93a29bb79b481a657f
3
+ size 7235
runs/Jan13_19-31-59_98f107f1aa39/events.out.tfevents.1705179585.98f107f1aa39.154347.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:771da7e33030b7c506cff2ebedb68fd5813ed7af9fee67bbca483a22096bc161
3
+ size 354
runs/Jan13_21-58-38_98f107f1aa39/events.out.tfevents.1705183294.98f107f1aa39.242277.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a28a9d3f0da08da4b1b876b1492bbd7d6affd9449a13c1543e6047e2e9802d5
3
+ size 6154
special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<unk>",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
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,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "bos_token": "<s>",
31
+ "chat_template": "{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '<|user|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'system' %}\n{{ '<|system|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'assistant' %}\n{{ '<|assistant|>\n' + message['content'] + eos_token }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '<|assistant|>' }}\n{% endif %}\n{% endfor %}",
32
+ "clean_up_tokenization_spaces": false,
33
+ "eos_token": "</s>",
34
+ "legacy": false,
35
+ "model_max_length": 2048,
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
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "train_loss": 1.1544433269235823,
4
+ "train_runtime": 5078.5964,
5
+ "train_samples": 29726,
6
+ "train_samples_per_second": 1.537,
7
+ "train_steps_per_second": 0.007
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 0.9961568024596464,
5
+ "eval_steps": 10,
6
+ "global_step": 36,
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.03,
13
+ "learning_rate": 3.9923893961834914e-05,
14
+ "loss": 1.3268,
15
+ "step": 1
16
+ },
17
+ {
18
+ "epoch": 0.14,
19
+ "learning_rate": 3.812615574073301e-05,
20
+ "loss": 1.2548,
21
+ "step": 5
22
+ },
23
+ {
24
+ "epoch": 0.28,
25
+ "learning_rate": 3.285575219373079e-05,
26
+ "loss": 1.1912,
27
+ "step": 10
28
+ },
29
+ {
30
+ "epoch": 0.28,
31
+ "eval_loss": 1.109934687614441,
32
+ "eval_runtime": 9.2946,
33
+ "eval_samples_per_second": 5.057,
34
+ "eval_steps_per_second": 1.291,
35
+ "step": 10
36
+ },
37
+ {
38
+ "epoch": 0.42,
39
+ "learning_rate": 2.5176380902050418e-05,
40
+ "loss": 1.1433,
41
+ "step": 15
42
+ },
43
+ {
44
+ "epoch": 0.55,
45
+ "learning_rate": 1.6527036446661396e-05,
46
+ "loss": 1.1238,
47
+ "step": 20
48
+ },
49
+ {
50
+ "epoch": 0.55,
51
+ "eval_loss": 1.065536379814148,
52
+ "eval_runtime": 9.2887,
53
+ "eval_samples_per_second": 5.06,
54
+ "eval_steps_per_second": 1.292,
55
+ "step": 20
56
+ },
57
+ {
58
+ "epoch": 0.69,
59
+ "learning_rate": 8.528471272979083e-06,
60
+ "loss": 1.1102,
61
+ "step": 25
62
+ },
63
+ {
64
+ "epoch": 0.83,
65
+ "learning_rate": 2.679491924311226e-06,
66
+ "loss": 1.1258,
67
+ "step": 30
68
+ },
69
+ {
70
+ "epoch": 0.83,
71
+ "eval_loss": 1.055001974105835,
72
+ "eval_runtime": 9.2909,
73
+ "eval_samples_per_second": 5.059,
74
+ "eval_steps_per_second": 1.292,
75
+ "step": 30
76
+ },
77
+ {
78
+ "epoch": 0.97,
79
+ "learning_rate": 7.61060381650891e-08,
80
+ "loss": 1.1272,
81
+ "step": 35
82
+ },
83
+ {
84
+ "epoch": 1.0,
85
+ "step": 36,
86
+ "total_flos": 1.0010669722946765e+17,
87
+ "train_loss": 1.1544433269235823,
88
+ "train_runtime": 5078.5964,
89
+ "train_samples_per_second": 1.537,
90
+ "train_steps_per_second": 0.007
91
+ }
92
+ ],
93
+ "logging_steps": 5,
94
+ "max_steps": 36,
95
+ "num_input_tokens_seen": 0,
96
+ "num_train_epochs": 1,
97
+ "save_steps": 20,
98
+ "total_flos": 1.0010669722946765e+17,
99
+ "train_batch_size": 6,
100
+ "trial_name": null,
101
+ "trial_params": null
102
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:69c81ac88e2869ccda740542c6596332cca7e1d65acbd50f130aca7858c31f79
3
+ size 4728