elichen3051 commited on
Commit
7056720
·
verified ·
1 Parent(s): af6b2b7

Model save

Browse files
Files changed (2) hide show
  1. README.md +166 -0
  2. generation_config.json +7 -0
README.md ADDED
@@ -0,0 +1,166 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: mistralai/Mistral-7B-v0.1
5
+ tags:
6
+ - axolotl
7
+ - generated_from_trainer
8
+ model-index:
9
+ - name: mistral-7B-v0.1-q-sparse-v6-on-slim-orca
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/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)
17
+ <details><summary>See axolotl config</summary>
18
+
19
+ axolotl version: `0.5.2`
20
+ ```yaml
21
+ base_model: mistralai/Mistral-7B-v0.1
22
+ model_type: AutoModelForCausalLM
23
+ tokenizer_config: Open-Orca/Mistral-7B-OpenOrca
24
+ tokenizer_type: AutoTokenizer
25
+ tokenizer_use_fast: false
26
+ resize_token_embeddings_to_32x: false
27
+
28
+ flash_attention: true
29
+ xformers_attention:
30
+
31
+ load_in_8bit: false
32
+ load_in_4bit: false
33
+ strict: false
34
+
35
+ chat_template: chatml
36
+ datasets:
37
+ - path: skymizer/Sonnet3.5-SlimOrcaDedupCleaned-train
38
+ type: chat_template
39
+ field_messages: messages
40
+
41
+ test_datasets:
42
+ - path: skymizer/Sonnet3.5-SlimOrcaDedupCleaned-test
43
+ type: chat_template
44
+ field_messages: messages
45
+ split: train
46
+
47
+ hf_use_auth_token: true
48
+ dataset_prepared_path: pretokenized/slim-orca
49
+ output_dir: ./exp_output_artifacts
50
+
51
+ sequence_len: 2048
52
+ sample_packing: true
53
+ pad_to_sequence_len: true
54
+
55
+ eval_sample_packing: false
56
+ # eval_causal_lm_metrics: ["perplexity"]
57
+
58
+ wandb_project: "axolotl_mistral_sft"
59
+ wandb_entity:
60
+ wandb_watch:
61
+ wandb_name: "mistral-7B-v0.1-q-spase-v6-on-slim-orca"
62
+ wandb_log_model:
63
+
64
+ gradient_accumulation_steps: 2
65
+ micro_batch_size: 16
66
+ eval_batch_size: 1
67
+ num_epochs: 1
68
+ optimizer: adamw_torch
69
+ lr_scheduler: cosine
70
+ learning_rate: 0.000005
71
+ warmup_ratio: 0.03
72
+ weight_decay: 0.0
73
+ adam_beta1: 0.9
74
+ adam_beta2: 0.95
75
+ adam_eps: 0.000001
76
+ max_grad_norm: 1.0
77
+
78
+ train_on_inputs: false
79
+ group_by_length: false
80
+ bf16: true
81
+ fp16:
82
+ tf32: false
83
+
84
+ hub_model_id: "skymizer/mistral-7B-v0.1-q-sparse-v6-on-slim-orca"
85
+
86
+ save_strategy: "steps"
87
+ save_steps: 50
88
+
89
+ gradient_checkpointing: true
90
+ gradient_checkpointing_kwargs:
91
+ use_reentrant: false
92
+ early_stopping_patience:
93
+ resume_from_checkpoint:
94
+ local_rank:
95
+ logging_steps: 1
96
+
97
+ eval_steps: 50
98
+ eval_table_size:
99
+ eval_max_new_tokens: 2048
100
+ debug:
101
+ deepspeed: deepspeed_configs/zero3_bf16.json
102
+ fsdp:
103
+ fsdp_config:
104
+ seed: 42
105
+
106
+ ```
107
+
108
+ </details><br>
109
+
110
+ # mistral-7B-v0.1-q-sparse-v6-on-slim-orca
111
+
112
+ This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on the None dataset.
113
+ It achieves the following results on the evaluation set:
114
+ - Loss: 2.7583
115
+
116
+ ## Model description
117
+
118
+ More information needed
119
+
120
+ ## Intended uses & limitations
121
+
122
+ More information needed
123
+
124
+ ## Training and evaluation data
125
+
126
+ More information needed
127
+
128
+ ## Training procedure
129
+
130
+ ### Training hyperparameters
131
+
132
+ The following hyperparameters were used during training:
133
+ - learning_rate: 5e-06
134
+ - train_batch_size: 16
135
+ - eval_batch_size: 1
136
+ - seed: 42
137
+ - distributed_type: multi-GPU
138
+ - num_devices: 4
139
+ - gradient_accumulation_steps: 2
140
+ - total_train_batch_size: 128
141
+ - total_eval_batch_size: 4
142
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.95) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
143
+ - lr_scheduler_type: cosine
144
+ - lr_scheduler_warmup_steps: 11
145
+ - num_epochs: 1
146
+
147
+ ### Training results
148
+
149
+ | Training Loss | Epoch | Step | Validation Loss |
150
+ |:-------------:|:------:|:----:|:---------------:|
151
+ | 11.586 | 0.0026 | 1 | 11.5238 |
152
+ | 7.2128 | 0.1277 | 50 | 7.1436 |
153
+ | 6.3388 | 0.2554 | 100 | 6.1285 |
154
+ | 5.4596 | 0.3831 | 150 | 5.3648 |
155
+ | 4.4506 | 0.5109 | 200 | 4.4040 |
156
+ | 3.5754 | 0.6386 | 250 | 3.5615 |
157
+ | 3.0413 | 0.7663 | 300 | 2.9740 |
158
+ | 2.8014 | 0.8940 | 350 | 2.7583 |
159
+
160
+
161
+ ### Framework versions
162
+
163
+ - Transformers 4.46.3
164
+ - Pytorch 2.5.1+cu124
165
+ - Datasets 3.1.0
166
+ - Tokenizers 0.20.3
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 1,
4
+ "do_sample": true,
5
+ "eos_token_id": 2,
6
+ "transformers_version": "4.46.3"
7
+ }