TearGosling commited on
Commit
bedbdc7
1 Parent(s): 018c3fc

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,178 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: qlora_outputs
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
+ datasets:
43
+ - path: /home/austin/disk1/summaries_fixed.jsonl
44
+ type: sharegpt
45
+ dataset_prepared_path: last_run_prepared
46
+ val_set_size: 0.01
47
+ output_dir: ./qlora_outputs
48
+
49
+ sequence_len: 8192
50
+ sample_packing: true
51
+ eval_sample_packing: false
52
+ pad_to_sequence_len: true
53
+
54
+ wandb_project: summarization-qlora
55
+ wandb_entity:
56
+ wandb_watch:
57
+ wandb_name: actual_run1
58
+ wandb_log_model:
59
+
60
+ #unsloth_cross_entropy_loss: true
61
+
62
+ gradient_accumulation_steps: 1
63
+ micro_batch_size: 1
64
+ num_epochs: 4
65
+ optimizer: adamw_bnb_8bit
66
+ lr_scheduler: cosine
67
+ learning_rate: 0.0002
68
+
69
+ train_on_inputs: false
70
+ group_by_length: false
71
+ bf16: auto
72
+ fp16:
73
+ tf32: false
74
+
75
+ gradient_checkpointing: true
76
+ gradient_checkpointing_kwargs:
77
+ use_reentrant: false
78
+ early_stopping_patience:
79
+ resume_from_checkpoint:
80
+ logging_steps: 1
81
+ xformers_attention: false
82
+ flash_attention: true
83
+
84
+ loss_watchdog_threshold: 5.0
85
+ loss_watchdog_patience: 3
86
+
87
+ warmup_steps: 25
88
+ evals_per_epoch: 4
89
+ eval_table_size:
90
+ saves_per_epoch: 4
91
+ debug:
92
+ deepspeed: ./deepspeed_configs/zero2.json
93
+ weight_decay: 0.0
94
+ fsdp:
95
+ # - full_shard
96
+ # - auto_wrap
97
+ fsdp_config:
98
+ # fsdp_limit_all_gathers: true
99
+ # fsdp_activation_checkpointing: true
100
+ # fsdp_sync_module_states: true
101
+ # fsdp_offload_params: false
102
+ # fsdp_use_orig_params: false
103
+ # fsdp_cpu_ram_efficient_loading: false
104
+ # fsdp_transformer_layer_cls_to_wrap: MistralDecoderLayer
105
+ # fsdp_state_dict_type: FULL_STATE_DICT
106
+ # fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
107
+ special_tokens:
108
+ pad_token: </s>
109
+
110
+ ```
111
+
112
+ </details><br>
113
+
114
+ # qlora_outputs
115
+
116
+ 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.
117
+ It achieves the following results on the evaluation set:
118
+ - Loss: 1.5617
119
+
120
+ ## Model description
121
+
122
+ More information needed
123
+
124
+ ## Intended uses & limitations
125
+
126
+ More information needed
127
+
128
+ ## Training and evaluation data
129
+
130
+ More information needed
131
+
132
+ ## Training procedure
133
+
134
+ ### Training hyperparameters
135
+
136
+ The following hyperparameters were used during training:
137
+ - learning_rate: 0.0002
138
+ - train_batch_size: 1
139
+ - eval_batch_size: 1
140
+ - seed: 42
141
+ - distributed_type: multi-GPU
142
+ - num_devices: 4
143
+ - total_train_batch_size: 4
144
+ - total_eval_batch_size: 4
145
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
146
+ - lr_scheduler_type: cosine
147
+ - lr_scheduler_warmup_steps: 25
148
+ - num_epochs: 4
149
+
150
+ ### Training results
151
+
152
+ | Training Loss | Epoch | Step | Validation Loss |
153
+ |:-------------:|:------:|:----:|:---------------:|
154
+ | 2.0177 | 0.0014 | 1 | 1.6514 |
155
+ | 1.6259 | 0.2507 | 177 | 1.2032 |
156
+ | 1.4232 | 0.5014 | 354 | 1.1897 |
157
+ | 1.6835 | 0.7521 | 531 | 1.1985 |
158
+ | 1.6514 | 1.0028 | 708 | 1.1874 |
159
+ | 1.4538 | 1.2365 | 885 | 1.2166 |
160
+ | 1.2421 | 1.4873 | 1062 | 1.2224 |
161
+ | 1.2844 | 1.7380 | 1239 | 1.2330 |
162
+ | 1.4152 | 1.9887 | 1416 | 1.2345 |
163
+ | 1.1668 | 2.2252 | 1593 | 1.3476 |
164
+ | 1.1249 | 2.4759 | 1770 | 1.3608 |
165
+ | 0.921 | 2.7266 | 1947 | 1.3793 |
166
+ | 0.7824 | 2.9773 | 2124 | 1.3906 |
167
+ | 1.1759 | 3.2040 | 2301 | 1.5438 |
168
+ | 0.6625 | 3.4547 | 2478 | 1.5644 |
169
+ | 0.8959 | 3.7054 | 2655 | 1.5617 |
170
+
171
+
172
+ ### Framework versions
173
+
174
+ - PEFT 0.12.0
175
+ - Transformers 4.44.0
176
+ - Pytorch 2.3.0+cu121
177
+ - Datasets 2.20.0
178
+ - 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
+ "gate_proj",
24
+ "up_proj",
25
+ "down_proj",
26
+ "o_proj",
27
+ "k_proj",
28
+ "q_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:6842a215f1e38f0301c0cee03ee41d3f80d8be8936c1527138a56b9c1e06e1d1
3
+ size 228153978
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