bastienp commited on
Commit
14ffac7
1 Parent(s): 02228e2

End of training

Browse files
Files changed (2) hide show
  1. README.md +148 -0
  2. adapter_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,148 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: peft
4
+ tags:
5
+ - axolotl
6
+ - generated_from_trainer
7
+ base_model: TinyLlama/TinyLlama-1.1B-Chat-v1.0
8
+ model-index:
9
+ - name: TinyLlamusk
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ [<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)
17
+ <details><summary>See axolotl config</summary>
18
+
19
+ axolotl version: `0.3.0`
20
+ ```yaml
21
+ base_model: TinyLlama/TinyLlama-1.1B-Chat-v1.0
22
+ model_type: LlamaForCausalLM
23
+ tokenizer_type: LlamaTokenizer
24
+ is_llama_derived_model: true
25
+ hub_model_id: TinyLlamusk
26
+
27
+ load_in_8bit: false
28
+ load_in_4bit: true
29
+ strict: false
30
+
31
+ datasets:
32
+ - path: lcama/elon-tweets
33
+ type: completion
34
+ dataset_prepared_path:
35
+ val_set_size: 0.02
36
+ output_dir: ./qlora-out
37
+
38
+ adapter: qlora
39
+ lora_model_dir:
40
+
41
+ sequence_len: 2048
42
+ sample_packing: true
43
+ pad_to_sequence_len: true
44
+
45
+ lora_r: 32
46
+ lora_alpha: 16
47
+ lora_dropout: 0.05
48
+ lora_target_modules:
49
+ lora_target_linear: true
50
+ lora_fan_in_fan_out:
51
+
52
+ wandb_project: axolotl-tinyllama
53
+ wandb_entity:
54
+ wandb_watch:
55
+ wandb_name:
56
+ wandb_log_model:
57
+
58
+ gradient_accumulation_steps: 2
59
+ micro_batch_size: 1
60
+ num_epochs: 4
61
+ optimizer: paged_adamw_32bit
62
+ lr_scheduler: cosine
63
+ learning_rate: 0.0002
64
+
65
+ train_on_inputs: false
66
+ group_by_length: false
67
+ bf16: false
68
+ fp16: true
69
+ tf32: false
70
+
71
+ gradient_checkpointing: true
72
+ early_stopping_patience:
73
+ resume_from_checkpoint:
74
+ local_rank:
75
+ logging_steps: 1
76
+ xformers_attention:
77
+ flash_attention: false
78
+
79
+ warmup_steps: 10
80
+ evals_per_epoch: 2
81
+ saves_per_epoch: 1
82
+ debug:
83
+ deepspeed:
84
+ weight_decay: 0.0
85
+ fsdp:
86
+ fsdp_config:
87
+ special_tokens:
88
+
89
+ ```
90
+
91
+ </details><br>
92
+
93
+ # TinyLlamusk
94
+
95
+ 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 None dataset.
96
+ It achieves the following results on the evaluation set:
97
+ - Loss: 4.7587
98
+
99
+ ## Model description
100
+
101
+ More information needed
102
+
103
+ ## Intended uses & limitations
104
+
105
+ More information needed
106
+
107
+ ## Training and evaluation data
108
+
109
+ More information needed
110
+
111
+ ## Training procedure
112
+
113
+ ### Training hyperparameters
114
+
115
+ The following hyperparameters were used during training:
116
+ - learning_rate: 0.0002
117
+ - train_batch_size: 1
118
+ - eval_batch_size: 1
119
+ - seed: 42
120
+ - gradient_accumulation_steps: 2
121
+ - total_train_batch_size: 2
122
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
123
+ - lr_scheduler_type: cosine
124
+ - lr_scheduler_warmup_steps: 10
125
+ - num_epochs: 4
126
+ - mixed_precision_training: Native AMP
127
+
128
+ ### Training results
129
+
130
+ | Training Loss | Epoch | Step | Validation Loss |
131
+ |:-------------:|:-----:|:----:|:---------------:|
132
+ | 6.8915 | 0.07 | 1 | 6.5485 |
133
+ | 6.1595 | 0.53 | 8 | 5.8708 |
134
+ | 5.361 | 1.03 | 16 | 5.2979 |
135
+ | 4.8874 | 1.57 | 24 | 5.0493 |
136
+ | 4.7517 | 2.07 | 32 | 4.9304 |
137
+ | 4.6544 | 2.6 | 40 | 4.8450 |
138
+ | 4.544 | 3.1 | 48 | 4.7767 |
139
+ | 4.4482 | 3.63 | 56 | 4.7587 |
140
+
141
+
142
+ ### Framework versions
143
+
144
+ - PEFT 0.7.2.dev0
145
+ - Transformers 4.37.0.dev0
146
+ - Pytorch 2.0.1+cu117
147
+ - Datasets 2.16.1
148
+ - Tokenizers 0.15.0
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:518ee3551aa2df1e215a1d1e5c79b2ecc4a94b3a57007e5475d0c9f073ee5ac9
3
+ size 101036253