lesso commited on
Commit
8940205
1 Parent(s): 7a1a314

End of training

Browse files
README.md ADDED
@@ -0,0 +1,158 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: llama3
4
+ base_model: tokyotech-llm/Llama-3-Swallow-8B-v0.1
5
+ tags:
6
+ - axolotl
7
+ - generated_from_trainer
8
+ model-index:
9
+ - name: 8cad90b9-d437-42b4-afb4-e51b67e64e50
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.4.1`
20
+ ```yaml
21
+ adapter: lora
22
+ base_model: tokyotech-llm/Llama-3-Swallow-8B-v0.1
23
+ bf16: false
24
+ chat_template: llama3
25
+ dataset_prepared_path: null
26
+ datasets:
27
+ - data_files:
28
+ - 3f72bc6633b04dfb_train_data.json
29
+ ds_type: json
30
+ format: custom
31
+ path: /workspace/input_data/3f72bc6633b04dfb_train_data.json
32
+ type:
33
+ field_input: titles
34
+ field_instruction: instructions
35
+ field_output: essays
36
+ format: '{instruction} {input}'
37
+ no_input_format: '{instruction}'
38
+ system_format: '{system}'
39
+ system_prompt: ''
40
+ debug: null
41
+ deepspeed: null
42
+ early_stopping_patience: null
43
+ eval_max_new_tokens: 128
44
+ eval_table_size: null
45
+ evals_per_epoch: 4
46
+ flash_attention: true
47
+ fp16: true
48
+ fsdp: null
49
+ fsdp_config: null
50
+ gradient_accumulation_steps: 4
51
+ gradient_checkpointing: true
52
+ group_by_length: false
53
+ hub_model_id: lesso/8cad90b9-d437-42b4-afb4-e51b67e64e50
54
+ hub_repo: null
55
+ hub_strategy: checkpoint
56
+ hub_token: null
57
+ learning_rate: 0.0002
58
+ load_in_4bit: false
59
+ load_in_8bit: false
60
+ local_rank: null
61
+ logging_steps: 1
62
+ lora_alpha: 16
63
+ lora_dropout: 0.05
64
+ lora_fan_in_fan_out: null
65
+ lora_model_dir: null
66
+ lora_r: 8
67
+ lora_target_linear: true
68
+ lr_scheduler: cosine
69
+ max_steps: 10
70
+ micro_batch_size: 1
71
+ mlflow_experiment_name: /tmp/3f72bc6633b04dfb_train_data.json
72
+ model_type: AutoModelForCausalLM
73
+ num_epochs: 1
74
+ optimizer: adamw_hf
75
+ output_dir: miner_id_24
76
+ pad_to_sequence_len: true
77
+ resume_from_checkpoint: null
78
+ s2_attention: null
79
+ sample_packing: false
80
+ saves_per_epoch: 4
81
+ sequence_len: 1024
82
+ special_tokens:
83
+ pad_token: <|end_of_text|>
84
+ strict: false
85
+ tf32: false
86
+ tokenizer_type: AutoTokenizer
87
+ train_on_inputs: false
88
+ trust_remote_code: true
89
+ val_set_size: 0.05
90
+ wandb_entity: null
91
+ wandb_mode: online
92
+ wandb_name: 8cad90b9-d437-42b4-afb4-e51b67e64e50
93
+ wandb_project: Gradients-On-Demand
94
+ wandb_run: your_name
95
+ wandb_runid: 8cad90b9-d437-42b4-afb4-e51b67e64e50
96
+ warmup_steps: 10
97
+ weight_decay: 0.0
98
+ xformers_attention: null
99
+
100
+ ```
101
+
102
+ </details><br>
103
+
104
+ # 8cad90b9-d437-42b4-afb4-e51b67e64e50
105
+
106
+ This model is a fine-tuned version of [tokyotech-llm/Llama-3-Swallow-8B-v0.1](https://huggingface.co/tokyotech-llm/Llama-3-Swallow-8B-v0.1) on the None dataset.
107
+ It achieves the following results on the evaluation set:
108
+ - Loss: 2.1792
109
+
110
+ ## Model description
111
+
112
+ More information needed
113
+
114
+ ## Intended uses & limitations
115
+
116
+ More information needed
117
+
118
+ ## Training and evaluation data
119
+
120
+ More information needed
121
+
122
+ ## Training procedure
123
+
124
+ ### Training hyperparameters
125
+
126
+ The following hyperparameters were used during training:
127
+ - learning_rate: 0.0002
128
+ - train_batch_size: 1
129
+ - eval_batch_size: 1
130
+ - seed: 42
131
+ - distributed_type: multi-GPU
132
+ - num_devices: 2
133
+ - gradient_accumulation_steps: 4
134
+ - total_train_batch_size: 8
135
+ - total_eval_batch_size: 2
136
+ - optimizer: Use OptimizerNames.ADAMW_HF with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
137
+ - lr_scheduler_type: cosine
138
+ - lr_scheduler_warmup_steps: 10
139
+ - training_steps: 10
140
+ - mixed_precision_training: Native AMP
141
+
142
+ ### Training results
143
+
144
+ | Training Loss | Epoch | Step | Validation Loss |
145
+ |:-------------:|:------:|:----:|:---------------:|
146
+ | 2.3204 | 0.0054 | 1 | 2.2009 |
147
+ | 2.2241 | 0.0162 | 3 | 2.1992 |
148
+ | 2.4093 | 0.0324 | 6 | 2.1916 |
149
+ | 2.2467 | 0.0486 | 9 | 2.1792 |
150
+
151
+
152
+ ### Framework versions
153
+
154
+ - PEFT 0.13.2
155
+ - Transformers 4.46.0
156
+ - Pytorch 2.5.0+cu124
157
+ - Datasets 3.0.1
158
+ - Tokenizers 0.20.1
adapter_config.json CHANGED
@@ -20,13 +20,13 @@
20
  "rank_pattern": {},
21
  "revision": null,
22
  "target_modules": [
23
- "gate_proj",
24
  "up_proj",
25
  "o_proj",
26
- "q_proj",
27
  "down_proj",
28
- "k_proj",
29
- "v_proj"
 
30
  ],
31
  "task_type": "CAUSAL_LM",
32
  "use_dora": false,
 
20
  "rank_pattern": {},
21
  "revision": null,
22
  "target_modules": [
23
+ "v_proj",
24
  "up_proj",
25
  "o_proj",
 
26
  "down_proj",
27
+ "q_proj",
28
+ "gate_proj",
29
+ "k_proj"
30
  ],
31
  "task_type": "CAUSAL_LM",
32
  "use_dora": false,
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6dd44e7c4608b202bc0f01c2f4a3f50f7ecd28e7fd03c4c0097240dbaadfaf2b
3
+ size 84047370
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:096d0d1306c228ce8c01652e2e9dc22fbfdf66e56fe1c3cc3c3f3341baa59e44
3
  size 83945296
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd8d602b8c8217d2bad01c439cf265d8636e230eef9088012f7f0abb56d28cf2
3
  size 83945296
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fd961290b5c1511222dcde7c5274eeb95e8e3143dbf3b403d7a4f50c5202a8b5
3
  size 6776
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:46bccd6964f4291101ddc4731a06bfcf4544fcbb608a5a0128bc2068c8364f7b
3
  size 6776