Delta-Vector commited on
Commit
51bde98
·
verified ·
1 Parent(s): b99952d

Model save

Browse files
Files changed (1) hide show
  1. README.md +186 -0
README.md ADDED
@@ -0,0 +1,186 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ base_model: NewEden/Hamanasu-KTO-4B
4
+ tags:
5
+ - axolotl
6
+ - generated_from_trainer
7
+ datasets:
8
+ - PocketDoc/Dans-Prosemaxx-Cowriter-3-S
9
+ - PocketDoc/Dans-Prosemaxx-Adventure
10
+ - PocketDoc/Dans-Failuremaxx-Adventure-3
11
+ - PocketDoc/Dans-Prosemaxx-InstructWriter-ZeroShot-2
12
+ - PocketDoc/Dans-Prosemaxx-InstructWriter-ZeroShot-3
13
+ - PocketDoc/Dans-Prosemaxx-InstructWriter-Continue-2
14
+ - PocketDoc/Dans-Prosemaxx-Instructwriter-Long
15
+ model-index:
16
+ - name: Hamanasu-4B-Adventure-Final-Hopefully
17
+ results: []
18
+ ---
19
+
20
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
21
+ should probably proofread and complete it, then remove this comment. -->
22
+
23
+ [<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
24
+ <details><summary>See axolotl config</summary>
25
+
26
+ axolotl version: `0.8.0.dev0`
27
+ ```yaml
28
+ base_model: NewEden/Hamanasu-KTO-4B
29
+ model_type: AutoModelForCausalLM
30
+ tokenizer_type: AutoTokenizer
31
+
32
+ hub_model_id: NewEden/Hamanasu-4B-Adventure-Final-Hopefully
33
+ hub_strategy: "all_checkpoints"
34
+ push_dataset_to_hub:
35
+ hf_use_auth_token: true
36
+
37
+ plugins:
38
+ - axolotl.integrations.liger.LigerPlugin
39
+ - axolotl.integrations.cut_cross_entropy.CutCrossEntropyPlugin
40
+ liger_rope: true
41
+ liger_rms_norm: true
42
+ liger_layer_norm: true
43
+ liger_glu_activation: true
44
+ liger_fused_linear_cross_entropy: false
45
+ cut_cross_entropy: true
46
+
47
+
48
+ load_in_8bit: false
49
+ load_in_4bit: false
50
+ strict: false
51
+
52
+ datasets:
53
+ - path: PocketDoc/Dans-Prosemaxx-Cowriter-3-S
54
+ type: dan-chat-advanced
55
+ - path: PocketDoc/Dans-Prosemaxx-Adventure
56
+ type: dan-chat-advanced
57
+ - path: PocketDoc/Dans-Failuremaxx-Adventure-3
58
+ type: dan-chat-advanced
59
+ - path: PocketDoc/Dans-Prosemaxx-InstructWriter-ZeroShot-2
60
+ type: dan-chat-advanced
61
+ - path: PocketDoc/Dans-Prosemaxx-InstructWriter-ZeroShot-3
62
+ type: dan-chat-advanced
63
+ - path: PocketDoc/Dans-Prosemaxx-InstructWriter-Continue-2
64
+ type: dan-chat-advanced
65
+ - path: PocketDoc/Dans-Prosemaxx-Instructwriter-Long
66
+ type: dan-chat-advanced
67
+ shuffle_merged_datasets: true
68
+ dataset_prepared_path: prepared_data
69
+ val_set_size: 0.01
70
+ output_dir: ./adventure
71
+
72
+ sequence_len: 32768
73
+ sample_packing: true
74
+ pad_to_sequence_len: true
75
+ eval_sample_packing: true
76
+
77
+
78
+ wandb_project: tavbussy
79
+ wandb_entity:
80
+ wandb_watch:
81
+ wandb_name: adventure-attempt-02
82
+ wandb_log_model:
83
+
84
+ gradient_accumulation_steps: 1
85
+ micro_batch_size: 4
86
+ num_epochs: 4
87
+ optimizer: adamw_bnb_8bit
88
+ lr_scheduler: cosine
89
+ learning_rate: 5e-6
90
+ max_grad_norm: 0.1
91
+
92
+ train_on_inputs: false
93
+ group_by_length: false
94
+ bf16: auto
95
+ fp16:
96
+ tf32: false
97
+
98
+ gradient_checkpointing: true
99
+ early_stopping_patience:
100
+ resume_from_checkpoint:
101
+ local_rank:
102
+ logging_steps: 1
103
+ xformers_attention:
104
+ flash_attention: true
105
+
106
+ warmup_steps: 25
107
+ evals_per_epoch: 4
108
+ eval_table_size:
109
+ eval_max_new_tokens: 128
110
+ saves_per_epoch: 2
111
+ debug:
112
+ deepspeed: ./deepspeed_configs/zero3_bf16.json
113
+ weight_decay: 0.02
114
+ fsdp:
115
+ fsdp_config:
116
+ special_tokens:
117
+ pad_token: <|finetune_right_pad_id|>
118
+
119
+ ```
120
+
121
+ </details><br>
122
+
123
+ # Hamanasu-4B-Adventure-Final-Hopefully
124
+
125
+ This model is a fine-tuned version of [NewEden/Hamanasu-KTO-4B](https://huggingface.co/NewEden/Hamanasu-KTO-4B) on the PocketDoc/Dans-Prosemaxx-Cowriter-3-S, the PocketDoc/Dans-Prosemaxx-Adventure, the PocketDoc/Dans-Failuremaxx-Adventure-3, the PocketDoc/Dans-Prosemaxx-InstructWriter-ZeroShot-2, the PocketDoc/Dans-Prosemaxx-InstructWriter-ZeroShot-3, the PocketDoc/Dans-Prosemaxx-InstructWriter-Continue-2 and the PocketDoc/Dans-Prosemaxx-Instructwriter-Long datasets.
126
+ It achieves the following results on the evaluation set:
127
+ - Loss: 2.4143
128
+
129
+ ## Model description
130
+
131
+ More information needed
132
+
133
+ ## Intended uses & limitations
134
+
135
+ More information needed
136
+
137
+ ## Training and evaluation data
138
+
139
+ More information needed
140
+
141
+ ## Training procedure
142
+
143
+ ### Training hyperparameters
144
+
145
+ The following hyperparameters were used during training:
146
+ - learning_rate: 5e-06
147
+ - train_batch_size: 4
148
+ - eval_batch_size: 4
149
+ - seed: 42
150
+ - distributed_type: multi-GPU
151
+ - num_devices: 4
152
+ - total_train_batch_size: 16
153
+ - total_eval_batch_size: 16
154
+ - optimizer: Use OptimizerNames.ADAMW_BNB with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
155
+ - lr_scheduler_type: cosine
156
+ - lr_scheduler_warmup_steps: 25
157
+ - num_epochs: 4.0
158
+
159
+ ### Training results
160
+
161
+ | Training Loss | Epoch | Step | Validation Loss |
162
+ |:-------------:|:------:|:----:|:---------------:|
163
+ | 2.5668 | 0.0068 | 1 | 2.5806 |
164
+ | 2.455 | 0.2534 | 37 | 2.4450 |
165
+ | 2.4115 | 0.5068 | 74 | 2.4323 |
166
+ | 2.3298 | 0.7603 | 111 | 2.4223 |
167
+ | 2.322 | 1.0137 | 148 | 2.4178 |
168
+ | 2.2661 | 1.2671 | 185 | 2.4178 |
169
+ | 2.2482 | 1.5205 | 222 | 2.4155 |
170
+ | 2.3707 | 1.7740 | 259 | 2.4115 |
171
+ | 2.293 | 2.0274 | 296 | 2.4132 |
172
+ | 2.3085 | 2.2808 | 333 | 2.4137 |
173
+ | 2.1902 | 2.5342 | 370 | 2.4123 |
174
+ | 2.216 | 2.7877 | 407 | 2.4112 |
175
+ | 2.3081 | 3.0411 | 444 | 2.4123 |
176
+ | 2.1989 | 3.2945 | 481 | 2.4142 |
177
+ | 2.2527 | 3.5479 | 518 | 2.4142 |
178
+ | 2.2419 | 3.8014 | 555 | 2.4143 |
179
+
180
+
181
+ ### Framework versions
182
+
183
+ - Transformers 4.50.0
184
+ - Pytorch 2.5.1+cu124
185
+ - Datasets 3.4.1
186
+ - Tokenizers 0.21.1