alpindale commited on
Commit
193db0d
·
verified ·
1 Parent(s): b03fed9

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,164 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: mistralai/Mistral-Nemo-Base-2407
3
+ library_name: peft
4
+ license: apache-2.0
5
+ tags:
6
+ - generated_from_trainer
7
+ model-index:
8
+ - name: home/austin/disk2/axolotl_storage/pyg3_qlora_2e-4
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ [<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)
16
+ <details><summary>See axolotl config</summary>
17
+
18
+ axolotl version: `0.4.1`
19
+ ```yaml
20
+ base_model: mistralai/Mistral-Nemo-Base-2407
21
+ model_type: AutoModelForCausalLM
22
+ tokenizer_type: AutoTokenizer
23
+
24
+ load_in_8bit: false
25
+ load_in_4bit: true
26
+ strict: false
27
+
28
+ plugins:
29
+ - axolotl.integrations.liger.LigerPlugin
30
+ liger_rope: true
31
+ liger_rms_norm: true
32
+ liger_swiglu: true
33
+ liger_fused_linear_cross_entropy: true
34
+
35
+ adapter: qlora
36
+ lora_r: 32
37
+ lora_alpha: 16
38
+ lora_dropout: 0.05
39
+ lora_target_linear: true
40
+ lora_fan_in_fan_out:
41
+
42
+ loraplus_lr_ratio: 2
43
+
44
+ chat_template: chatml
45
+
46
+ datasets:
47
+ - path: /home/austin/disk2/axolotl_data/fixed_pyg3.jsonl
48
+ type: sharegpt
49
+ conversation: chatml
50
+
51
+ dataset_prepared_path: /home/austin/disk2/axolotl_data/data_tokenized
52
+ val_set_size: 0.01
53
+ output_dir: /home/austin/disk2/axolotl_storage/pyg3_qlora_2e-4
54
+
55
+ sequence_len: 8192
56
+ sample_packing: true
57
+ eval_sample_packing: true
58
+ pad_to_sequence_len: true
59
+
60
+ wandb_project: pyg3-qlora
61
+ wandb_entity:
62
+ wandb_watch:
63
+ wandb_name: 1e-5
64
+ wandb_log_model:
65
+
66
+ #unsloth_cross_entropy_loss: true
67
+
68
+ gradient_accumulation_steps: 1
69
+ micro_batch_size: 3
70
+ num_epochs: 1
71
+ optimizer: paged_adamw_8bit
72
+ lr_scheduler: cosine
73
+ learning_rate: 0.0002
74
+
75
+ train_on_inputs: false
76
+ group_by_length: false
77
+ bf16: auto
78
+ fp16:
79
+ tf32: false
80
+
81
+ gradient_checkpointing: true
82
+ gradient_checkpointing_kwargs:
83
+ use_reentrant: false
84
+ early_stopping_patience:
85
+ resume_from_checkpoint:
86
+ logging_steps: 1
87
+ xformers_attention:
88
+ flash_attention: true
89
+
90
+ warmup_steps: 100
91
+ evals_per_epoch: 10
92
+ eval_table_size:
93
+ saves_per_epoch: 10
94
+ debug:
95
+ deepspeed: deepspeed_configs/zero2.json
96
+ weight_decay: 0.01
97
+ fsdp:
98
+ fsdp_config:
99
+ special_tokens:
100
+ pad_token: </s>
101
+
102
+ ```
103
+
104
+ </details><br>
105
+
106
+ # home/austin/disk2/axolotl_storage/pyg3_qlora_2e-4
107
+
108
+ This model is a fine-tuned version of [mistralai/Mistral-Nemo-Base-2407](https://huggingface.co/mistralai/Mistral-Nemo-Base-2407) on the None dataset.
109
+ It achieves the following results on the evaluation set:
110
+ - Loss: 0.8024
111
+
112
+ ## Model description
113
+
114
+ More information needed
115
+
116
+ ## Intended uses & limitations
117
+
118
+ More information needed
119
+
120
+ ## Training and evaluation data
121
+
122
+ More information needed
123
+
124
+ ## Training procedure
125
+
126
+ ### Training hyperparameters
127
+
128
+ The following hyperparameters were used during training:
129
+ - learning_rate: 0.0002
130
+ - train_batch_size: 3
131
+ - eval_batch_size: 3
132
+ - seed: 42
133
+ - distributed_type: multi-GPU
134
+ - num_devices: 8
135
+ - total_train_batch_size: 24
136
+ - total_eval_batch_size: 24
137
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
138
+ - lr_scheduler_type: cosine
139
+ - lr_scheduler_warmup_steps: 100
140
+ - num_epochs: 1
141
+
142
+ ### Training results
143
+
144
+ | Training Loss | Epoch | Step | Validation Loss |
145
+ |:-------------:|:------:|:----:|:---------------:|
146
+ | 1.8656 | 0.0006 | 1 | 1.1181 |
147
+ | 1.5716 | 0.1004 | 175 | 0.8479 |
148
+ | 1.6573 | 0.2008 | 350 | 0.8308 |
149
+ | 1.8387 | 0.3012 | 525 | 0.8230 |
150
+ | 1.5855 | 0.4016 | 700 | 0.8167 |
151
+ | 1.7139 | 0.5020 | 875 | 0.8123 |
152
+ | 1.5684 | 0.6024 | 1050 | 0.8087 |
153
+ | 1.6986 | 0.7028 | 1225 | 0.8055 |
154
+ | 1.6505 | 0.8032 | 1400 | 0.8035 |
155
+ | 1.6028 | 0.9036 | 1575 | 0.8024 |
156
+
157
+
158
+ ### Framework versions
159
+
160
+ - PEFT 0.12.0
161
+ - Transformers 4.44.0
162
+ - Pytorch 2.4.0+cu121
163
+ - Datasets 2.20.0
164
+ - Tokenizers 0.19.1
adapter_config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "mistralai/Mistral-Nemo-Base-2407",
5
+ "bias": "none",
6
+ "fan_in_fan_out": null,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layer_replication": null,
10
+ "layers_pattern": null,
11
+ "layers_to_transform": null,
12
+ "loftq_config": {},
13
+ "lora_alpha": 16,
14
+ "lora_dropout": 0.05,
15
+ "megatron_config": null,
16
+ "megatron_core": "megatron.core",
17
+ "modules_to_save": null,
18
+ "peft_type": "LORA",
19
+ "r": 32,
20
+ "rank_pattern": {},
21
+ "revision": null,
22
+ "target_modules": [
23
+ "o_proj",
24
+ "gate_proj",
25
+ "q_proj",
26
+ "down_proj",
27
+ "up_proj",
28
+ "k_proj",
29
+ "v_proj"
30
+ ],
31
+ "task_type": "CAUSAL_LM",
32
+ "use_dora": false,
33
+ "use_rslora": false
34
+ }
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f7da3153df1d65ac227e717f10dc6e338b9779aedd833369b7354fa7767ee22
3
+ size 228154172
config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "mistralai/Mistral-Nemo-Base-2407",
3
+ "architectures": [
4
+ "MistralForCausalLM"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 1,
8
+ "eos_token_id": 2,
9
+ "head_dim": 128,
10
+ "hidden_act": "silu",
11
+ "hidden_size": 5120,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 14336,
14
+ "max_position_embeddings": 1024000,
15
+ "model_type": "mistral",
16
+ "num_attention_heads": 32,
17
+ "num_hidden_layers": 40,
18
+ "num_key_value_heads": 8,
19
+ "quantization_config": {
20
+ "_load_in_4bit": true,
21
+ "_load_in_8bit": false,
22
+ "bnb_4bit_compute_dtype": "bfloat16",
23
+ "bnb_4bit_quant_storage": "bfloat16",
24
+ "bnb_4bit_quant_type": "nf4",
25
+ "bnb_4bit_use_double_quant": true,
26
+ "llm_int8_enable_fp32_cpu_offload": false,
27
+ "llm_int8_has_fp16_weight": false,
28
+ "llm_int8_skip_modules": null,
29
+ "llm_int8_threshold": 6.0,
30
+ "load_in_4bit": true,
31
+ "load_in_8bit": false,
32
+ "quant_method": "bitsandbytes"
33
+ },
34
+ "rms_norm_eps": 1e-05,
35
+ "rope_theta": 1000000.0,
36
+ "sliding_window": null,
37
+ "tie_word_embeddings": false,
38
+ "torch_dtype": "bfloat16",
39
+ "transformers_version": "4.44.0",
40
+ "use_cache": false,
41
+ "vocab_size": 131072
42
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<unk>",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
The diff for this file is too large to render. See raw diff