jeiku commited on
Commit
b74f29c
1 Parent(s): b13b15c

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +171 -0
README.md ADDED
@@ -0,0 +1,171 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ base_model: Dans-DiscountModels/Mistral-NeMo-Minitron-8B-Base-ChatML
4
+ tags:
5
+ - generated_from_trainer
6
+ model-index:
7
+ - name: outputs/out
8
+ results: []
9
+ ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ [<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)
15
+ <details><summary>See axolotl config</summary>
16
+
17
+ axolotl version: `0.4.1`
18
+ ```yaml
19
+ base_model: Dans-DiscountModels/Mistral-NeMo-Minitron-8B-Base-ChatML
20
+ model_type: AutoModelForCausalLM
21
+ tokenizer_type: AutoTokenizer
22
+
23
+ load_in_8bit: false
24
+ load_in_4bit: false
25
+ strict: false
26
+
27
+ datasets:
28
+ - path: PocketDoc/Dans-MemoryCore-CoreCurriculum-Small
29
+ type: sharegpt
30
+ conversation: chatml
31
+ - path: NewEden/Kalo-Opus-Instruct-22k-Refusal-Murdered
32
+ type: sharegpt
33
+ conversation: chatml
34
+ - path: Epiculous/Synthstruct-Gens-v1.1-Filtered-n-Cleaned
35
+ type: sharegpt
36
+ conversation: chatml
37
+ - path: NewEden/Gryphe-Sonnet-3.5-35k-Subset
38
+ type: sharegpt
39
+ conversation: chatml
40
+ - path: jeiku/Nitral_Medical_Dialog_Fixed
41
+ type: sharegpt
42
+ conversation: chatml
43
+
44
+ chat_template: chatml
45
+
46
+ val_set_size: 0.01
47
+ output_dir: ./outputs/out
48
+
49
+ adapter:
50
+ lora_r:
51
+ lora_alpha:
52
+ lora_dropout:
53
+ lora_target_linear:
54
+
55
+ sequence_len: 8192
56
+ # sequence_len: 32768
57
+ sample_packing: true
58
+ eval_sample_packing: false
59
+ pad_to_sequence_len: true
60
+
61
+ plugins:
62
+ - axolotl.integrations.liger.LigerPlugin
63
+ liger_rope: true
64
+ liger_rms_norm: true
65
+ liger_swiglu: true
66
+ liger_fused_linear_cross_entropy: true
67
+
68
+ wandb_project: New8B
69
+ wandb_entity:
70
+ wandb_watch:
71
+ wandb_name: New8B
72
+ wandb_log_model:
73
+
74
+ gradient_accumulation_steps: 32
75
+ micro_batch_size: 1
76
+ num_epochs: 2
77
+ optimizer: adamw_bnb_8bit
78
+ lr_scheduler: cosine
79
+ learning_rate: 0.00001
80
+ weight_decay: 0.05
81
+
82
+ train_on_inputs: false
83
+ group_by_length: false
84
+ bf16: auto
85
+ fp16:
86
+ tf32: true
87
+
88
+ gradient_checkpointing: true
89
+ early_stopping_patience:
90
+ resume_from_checkpoint:
91
+ local_rank:
92
+ logging_steps: 1
93
+ xformers_attention:
94
+ flash_attention: true
95
+
96
+ warmup_ratio: 0.1
97
+ evals_per_epoch: 4
98
+ eval_table_size:
99
+ eval_max_new_tokens: 128
100
+ saves_per_epoch: 2
101
+
102
+ debug:
103
+ deepspeed:
104
+ fsdp:
105
+ fsdp_config:
106
+
107
+ special_tokens:
108
+ pad_token: <pad>
109
+
110
+
111
+ ```
112
+
113
+ </details><br>
114
+
115
+ # outputs/out
116
+
117
+ This model is a fine-tuned version of [Dans-DiscountModels/Mistral-NeMo-Minitron-8B-Base-ChatML](https://huggingface.co/Dans-DiscountModels/Mistral-NeMo-Minitron-8B-Base-ChatML) on the None dataset.
118
+ It achieves the following results on the evaluation set:
119
+ - Loss: 1.0138
120
+
121
+ ## Model description
122
+
123
+ More information needed
124
+
125
+ ## Intended uses & limitations
126
+
127
+ More information needed
128
+
129
+ ## Training and evaluation data
130
+
131
+ More information needed
132
+
133
+ ## Training procedure
134
+
135
+ ### Training hyperparameters
136
+
137
+ The following hyperparameters were used during training:
138
+ - learning_rate: 1e-05
139
+ - train_batch_size: 1
140
+ - eval_batch_size: 1
141
+ - seed: 42
142
+ - distributed_type: multi-GPU
143
+ - num_devices: 2
144
+ - gradient_accumulation_steps: 32
145
+ - total_train_batch_size: 64
146
+ - total_eval_batch_size: 2
147
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
148
+ - lr_scheduler_type: cosine
149
+ - lr_scheduler_warmup_steps: 23
150
+ - num_epochs: 2
151
+
152
+ ### Training results
153
+
154
+ | Training Loss | Epoch | Step | Validation Loss |
155
+ |:-------------:|:------:|:----:|:---------------:|
156
+ | 1.4041 | 0.0084 | 1 | 1.5854 |
157
+ | 1.0445 | 0.2534 | 30 | 1.1313 |
158
+ | 0.9479 | 0.5069 | 60 | 1.0488 |
159
+ | 0.9697 | 0.7603 | 90 | 1.0288 |
160
+ | 0.9126 | 1.0090 | 120 | 1.0193 |
161
+ | 0.9006 | 1.2621 | 150 | 1.0156 |
162
+ | 0.9038 | 1.5153 | 180 | 1.0140 |
163
+ | 0.9095 | 1.7685 | 210 | 1.0138 |
164
+
165
+
166
+ ### Framework versions
167
+
168
+ - Transformers 4.45.0.dev0
169
+ - Pytorch 2.4.0+cu121
170
+ - Datasets 2.21.0
171
+ - Tokenizers 0.19.1