hardlyworking commited on
Commit
60e8893
·
verified ·
1 Parent(s): 35295ec

End of training

Browse files
Files changed (2) hide show
  1. README.md +176 -0
  2. generation_config.json +9 -0
README.md ADDED
@@ -0,0 +1,176 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: cc-by-nc-4.0
4
+ base_model: Salesforce/xgen-small-4B-instruct-r
5
+ tags:
6
+ - axolotl
7
+ - generated_from_trainer
8
+ datasets:
9
+ - hardlyworking/HardlyRPv2
10
+ model-index:
11
+ - name: HoldMy4B
12
+ results: []
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+ [<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)
19
+ <details><summary>See axolotl config</summary>
20
+
21
+ axolotl version: `0.10.0`
22
+ ```yaml
23
+ base_model: Salesforce/xgen-small-4B-instruct-r
24
+
25
+ load_in_8bit: false
26
+ load_in_4bit: false
27
+ strict: false
28
+
29
+ chat_template: chatml
30
+ datasets:
31
+ - path: hardlyworking/HardlyRPv2
32
+ type: chat_template
33
+ split: train
34
+ field_messages: conversations
35
+ message_property_mappings:
36
+ role: from
37
+ content: value
38
+ val_set_size: 0.1
39
+ output_dir: ./outputs/out
40
+ dataset_prepared_path: last_run_prepared
41
+ shuffle_merged_datasets: true
42
+
43
+ hub_model_id: hardlyworking/HoldMy4B
44
+ hub_strategy: "all_checkpoints"
45
+ push_dataset_to_hub:
46
+ hf_use_auth_token: true
47
+
48
+ plugins:
49
+ - axolotl.integrations.liger.LigerPlugin
50
+ - axolotl.integrations.cut_cross_entropy.CutCrossEntropyPlugin
51
+ liger_rope: true
52
+ liger_rms_norm: true
53
+ liger_layer_norm: true
54
+ liger_glu_activation: true
55
+ liger_fused_linear_cross_entropy: false
56
+ cut_cross_entropy: true
57
+
58
+ sequence_len: 32768
59
+ sample_packing: true
60
+ eval_sample_packing: true
61
+ pad_to_sequence_len: true
62
+
63
+ wandb_project: Xgen4B
64
+ wandb_entity:
65
+ wandb_watch:
66
+ wandb_name: Xgen4B
67
+ wandb_log_model:
68
+
69
+ evals_per_epoch: 8
70
+ eval_table_size:
71
+ eval_max_new_tokens: 128
72
+
73
+ gradient_accumulation_steps: 2
74
+ micro_batch_size: 2
75
+ num_epochs: 2
76
+ optimizer: adamw_bnb_8bit
77
+ lr_scheduler: cosine
78
+ learning_rate: 1e-5
79
+
80
+ train_on_inputs: false
81
+ group_by_length: false
82
+ bf16: auto
83
+ fp16:
84
+ tf32: false
85
+
86
+ gradient_checkpointing: offload
87
+ gradient_checkpointing_kwargs:
88
+ use_reentrant: false
89
+ early_stopping_patience:
90
+ resume_from_checkpoint:
91
+ local_rank:
92
+ logging_steps: 1
93
+ xformers_attention:
94
+ flash_attention: true
95
+ s2_attention:
96
+
97
+ deepspeed:
98
+
99
+ warmup_ratio: 0.05
100
+ saves_per_epoch: 1
101
+ debug:
102
+ weight_decay: 0.01
103
+ fsdp:
104
+ fsdp_config:
105
+ special_tokens:
106
+ pad_token:
107
+ ```
108
+
109
+ </details><br>
110
+
111
+ # HoldMy4B
112
+
113
+ This model is a fine-tuned version of [Salesforce/xgen-small-4B-instruct-r](https://huggingface.co/Salesforce/xgen-small-4B-instruct-r) on the hardlyworking/HardlyRPv2 dataset.
114
+ It achieves the following results on the evaluation set:
115
+ - Loss: 2.1637
116
+
117
+ ## Model description
118
+
119
+ More information needed
120
+
121
+ ## Intended uses & limitations
122
+
123
+ More information needed
124
+
125
+ ## Training and evaluation data
126
+
127
+ More information needed
128
+
129
+ ## Training procedure
130
+
131
+ ### Training hyperparameters
132
+
133
+ The following hyperparameters were used during training:
134
+ - learning_rate: 1e-05
135
+ - train_batch_size: 2
136
+ - eval_batch_size: 2
137
+ - seed: 42
138
+ - distributed_type: multi-GPU
139
+ - num_devices: 2
140
+ - gradient_accumulation_steps: 2
141
+ - total_train_batch_size: 8
142
+ - total_eval_batch_size: 4
143
+ - optimizer: Use OptimizerNames.ADAMW_BNB with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
144
+ - lr_scheduler_type: cosine
145
+ - lr_scheduler_warmup_steps: 24
146
+ - training_steps: 480
147
+
148
+ ### Training results
149
+
150
+ | Training Loss | Epoch | Step | Validation Loss |
151
+ |:-------------:|:-----:|:----:|:---------------:|
152
+ | No log | 0 | 0 | 2.6420 |
153
+ | 2.0119 | 0.125 | 30 | 2.2105 |
154
+ | 1.8963 | 0.25 | 60 | 2.1865 |
155
+ | 1.8623 | 0.375 | 90 | 2.1787 |
156
+ | 1.8528 | 0.5 | 120 | 2.1746 |
157
+ | 1.8784 | 0.625 | 150 | 2.1706 |
158
+ | 1.9961 | 0.75 | 180 | 2.1686 |
159
+ | 1.8748 | 0.875 | 210 | 2.1672 |
160
+ | 2.0385 | 1.0 | 240 | 2.1657 |
161
+ | 1.9327 | 1.125 | 270 | 2.1646 |
162
+ | 1.8509 | 1.25 | 300 | 2.1645 |
163
+ | 1.8279 | 1.375 | 330 | 2.1640 |
164
+ | 1.8271 | 1.5 | 360 | 2.1638 |
165
+ | 1.8589 | 1.625 | 390 | 2.1637 |
166
+ | 1.9824 | 1.75 | 420 | 2.1637 |
167
+ | 1.8668 | 1.875 | 450 | 2.1637 |
168
+ | 2.0332 | 2.0 | 480 | 2.1637 |
169
+
170
+
171
+ ### Framework versions
172
+
173
+ - Transformers 4.52.3
174
+ - Pytorch 2.6.0+cu124
175
+ - Datasets 3.6.0
176
+ - Tokenizers 0.21.1
generation_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 100257,
4
+ "do_sample": true,
5
+ "eos_token_id": 100265,
6
+ "pad_token_id": 100257,
7
+ "transformers_version": "4.52.3",
8
+ "use_cache": false
9
+ }