Views
No views yet


0.4.01base_model: cognitivecomputations/mixtral-1x22b-base
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4
5# trust_remote_code: true
6
7# load_in_8bit: true
8# load_in_4bit: true
9# strict: false
10
11datasets:
12 - path: /workspace/datasets/dolphin-2.9/dolphin201-sharegpt2.jsonl
13 type: sharegpt
14 conversation: chatml
15 - path: /workspace/datasets/dolphin-2.9/dolphin-coder-translate-sharegpt2.jsonl
16 type: sharegpt
17 conversation: chatml
18 - path: /workspace/datasets/dolphin-2.9/dolphin-coder-codegen-sharegpt2.jsonl
19 type: sharegpt
20 conversation: chatml
21 - path: /workspace/datasets/dolphin-2.9/m-a-p_Code-Feedback-sharegpt-unfiltered.jsonl
22 type: sharegpt
23 conversation: chatml
24 - path: /workspace/datasets/dolphin-2.9/m-a-p_CodeFeedback-Filtered-Instruction-sharegpt-unfiltered.jsonl
25 type: sharegpt
26 conversation: chatml
27 - path: /workspace/datasets/dolphin-2.9/not_samantha_norefusals.jsonl
28 type: sharegpt
29 conversation: chatml
30 - path: /workspace/datasets/dolphin-2.9/Orca-Math-resort-unfiltered.jsonl
31 type: sharegpt
32 conversation: chatml
33 - path: /workspace/datasets/dolphin-2.9/agent_instruct_react_unfiltered.jsonl
34 type: sharegpt
35 conversation: chatml
36 - path: /workspace/datasets/dolphin-2.9/toolbench_instruct_j1s1_3k_unfiltered.jsonl
37 type: sharegpt
38 conversation: chatml
39 - path: /workspace/datasets/dolphin-2.9/toolbench_negative_unfiltered.jsonl
40 type: sharegpt
41 conversation: chatml
42 - path: /workspace/datasets/dolphin-2.9/toolbench_react_10p_unfiltered.jsonl
43 type: sharegpt
44 conversation: chatml
45 - path: /workspace/datasets/dolphin-2.9/toolbench_tflan_cot_30p_unfiltered.jsonl
46 type: sharegpt
47 conversation: chatml
48 - path: /workspace/datasets/dolphin-2.9/openhermes200k_unfiltered.jsonl
49 type: sharegpt
50 conversation: chatml
51
52chat_template: chatml
53dataset_prepared_path: yi34b-prepared
54val_set_size: 0.01
55output_dir: ./1x22b-out
56
57# adapter: qlora
58# lora_r: 16
59# lora_alpha: 16
60# lora_modules_to_save: [embed_tokens, lm_head]
61# lora_dropout: 0.05
62# lora_target_linear: true
63
64# unfrozen_parameters:
65# - ^lm_head.weight$
66# - ^model.embed_tokens.weight$
67# # input_layernorm layers
68# - model.layers.0.input_layernorm
69# - model.layers.1.input_layernorm
70# - model.layers.2.input_layernorm
71# - model.layers.3.input_layernorm
72# - model.layers.4.input_layernorm
73# - model.layers.5.input_layernorm
74# - model.layers.6.input_layernorm
75# - model.layers.7.input_layernorm
76# - model.layers.8.input_layernorm
77# - model.layers.9.input_layernorm
78# - model.layers.10.input_layernorm
79# - model.layers.11.input_layernorm
80# - model.layers.12.input_layernorm
81# - model.layers.13.input_layernorm
82# - model.layers.14.input_layernorm
83# - model.layers.15.input_layernorm
84# - model.layers.16.input_layernorm
85# - model.layers.17.input_layernorm
86# - model.layers.18.input_layernorm
87# - model.layers.19.input_layernorm
88# - model.layers.20.input_layernorm
89# - model.layers.21.input_layernorm
90# - model.layers.22.input_layernorm
91# - model.layers.23.input_layernorm
92# # lm_head layers
93# # mlp.down_proj layers
94# - model.layers.17.mlp.down_proj
95# - model.layers.18.mlp.down_proj
96# - model.layers.19.mlp.down_proj
97# - model.layers.20.mlp.down_proj
98# - model.layers.21.mlp.down_proj
99# - model.layers.22.mlp.down_proj
100# - model.layers.23.mlp.down_proj
101# - model.layers.24.mlp.down_proj
102# - model.layers.25.mlp.down_proj
103# - model.layers.26.mlp.down_proj
104# - model.layers.27.mlp.down_proj
105# - model.layers.28.mlp.down_proj
106# - model.layers.29.mlp.down_proj
107# - model.layers.30.mlp.down_proj
108# - model.layers.31.mlp.down_proj
109# - model.layers.32.mlp.down_proj
110# - model.layers.33.mlp.down_proj
111# - model.layers.34.mlp.down_proj
112# - model.layers.35.mlp.down_proj
113# - model.layers.36.mlp.down_proj
114# - model.layers.37.mlp.down_proj
115# - model.layers.38.mlp.down_proj
116# - model.layers.39.mlp.down_proj
117# - model.layers.40.mlp.down_proj
118# # mlp.gate_proj layers
119# - model.layers.51.mlp.gate_proj
120# - model.layers.50.mlp.gate_proj
121# - model.layers.53.mlp.gate_proj
122# - model.layers.52.mlp.gate_proj
123# - model.layers.49.mlp.gate_proj
124# - model.layers.45.mlp.gate_proj
125# - model.layers.46.mlp.gate_proj
126# - model.layers.47.mlp.gate_proj
127# - model.layers.57.mlp.gate_proj
128# - model.layers.48.mlp.gate_proj
129# - model.layers.56.mlp.gate_proj
130# - model.layers.41.mlp.gate_proj
131# - model.layers.54.mlp.gate_proj
132# - model.layers.43.mlp.gate_proj
133# - model.layers.44.mlp.gate_proj
134# - model.layers.60.mlp.gate_proj
135# - model.layers.55.mlp.gate_proj
136# - model.layers.40.mlp.gate_proj
137# - model.layers.42.mlp.gate_proj
138# - model.layers.58.mlp.gate_proj
139# - model.layers.36.mlp.gate_proj
140# - model.layers.37.mlp.gate_proj
141# - model.layers.38.mlp.gate_proj
142# - model.layers.39.mlp.gate_proj
143# # mlp.up_proj layers
144# - model.layers.50.mlp.up_proj
145# - model.layers.51.mlp.up_proj
146# - model.layers.41.mlp.up_proj
147# - model.layers.49.mlp.up_proj
148# - model.layers.43.mlp.up_proj
149# - model.layers.44.mlp.up_proj
150# - model.layers.40.mlp.up_proj
151# - model.layers.45.mlp.up_proj
152# - model.layers.47.mlp.up_proj
153# - model.layers.48.mlp.up_proj
154# - model.layers.46.mlp.up_proj
155# - model.layers.42.mlp.up_proj
156# - model.layers.39.mlp.up_proj
157# - model.layers.36.mlp.up_proj
158# - model.layers.37.mlp.up_proj
159# - model.layers.38.mlp.up_proj
160# - model.layers.56.mlp.up_proj
161# - model.layers.57.mlp.up_proj
162# - model.layers.53.mlp.up_proj
163# - model.layers.31.mlp.up_proj
164# - model.layers.32.mlp.up_proj
165# - model.layers.34.mlp.up_proj
166# - model.layers.35.mlp.up_proj
167# - model.layers.33.mlp.up_proj
168# # model.embed_tokens layers
169# # model.norm layers
170# # post_attention_layernorm layers
171# - model.layers.0.post_attention_layernorm
172# - model.layers.1.post_attention_layernorm
173# - model.layers.2.post_attention_layernorm
174# - model.layers.3.post_attention_layernorm
175# - model.layers.4.post_attention_layernorm
176# - model.layers.5.post_attention_layernorm
177# - model.layers.6.post_attention_layernorm
178# - model.layers.7.post_attention_layernorm
179# - model.layers.8.post_attention_layernorm
180# - model.layers.9.post_attention_layernorm
181# - model.layers.10.post_attention_layernorm
182# - model.layers.11.post_attention_layernorm
183# - model.layers.12.post_attention_layernorm
184# - model.layers.13.post_attention_layernorm
185# - model.layers.14.post_attention_layernorm
186# - model.layers.15.post_attention_layernorm
187# - model.layers.16.post_attention_layernorm
188# - model.layers.17.post_attention_layernorm
189# - model.layers.18.post_attention_layernorm
190# - model.layers.19.post_attention_layernorm
191# - model.layers.20.post_attention_layernorm
192# - model.layers.21.post_attention_layernorm
193# - model.layers.22.post_attention_layernorm
194# - model.layers.23.post_attention_layernorm
195# # self_attn.k_proj layers
196# - model.layers.42.self_attn.k_proj
197# - model.layers.41.self_attn.k_proj
198# - model.layers.39.self_attn.k_proj
199# - model.layers.35.self_attn.k_proj
200# - model.layers.28.self_attn.k_proj
201# - model.layers.79.self_attn.k_proj
202# - model.layers.43.self_attn.k_proj
203# - model.layers.32.self_attn.k_proj
204# - model.layers.73.self_attn.k_proj
205# - model.layers.31.self_attn.k_proj
206# - model.layers.29.self_attn.k_proj
207# - model.layers.76.self_attn.k_proj
208# - model.layers.30.self_attn.k_proj
209# - model.layers.40.self_attn.k_proj
210# - model.layers.33.self_attn.k_proj
211# - model.layers.78.self_attn.k_proj
212# - model.layers.34.self_attn.k_proj
213# - model.layers.37.self_attn.k_proj
214# - model.layers.45.self_attn.k_proj
215# - model.layers.44.self_attn.k_proj
216# - model.layers.71.self_attn.k_proj
217# - model.layers.26.self_attn.k_proj
218# - model.layers.74.self_attn.k_proj
219# - model.layers.27.self_attn.k_proj
220# # self_attn.o_proj layers
221# - model.layers.35.self_attn.o_proj
222# - model.layers.34.self_attn.o_proj
223# - model.layers.37.self_attn.o_proj
224# - model.layers.33.self_attn.o_proj
225# - model.layers.31.self_attn.o_proj
226# - model.layers.27.self_attn.o_proj
227# - model.layers.38.self_attn.o_proj
228# - model.layers.24.self_attn.o_proj
229# - model.layers.39.self_attn.o_proj
230# - model.layers.43.self_attn.o_proj
231# - model.layers.29.self_attn.o_proj
232# - model.layers.0.self_attn.o_proj
233# - model.layers.50.self_attn.o_proj
234# - model.layers.32.self_attn.o_proj
235# - model.layers.45.self_attn.o_proj
236# - model.layers.30.self_attn.o_proj
237# - model.layers.60.self_attn.o_proj
238# - model.layers.23.self_attn.o_proj
239# - model.layers.18.self_attn.o_proj
240# - model.layers.67.self_attn.o_proj
241# - model.layers.57.self_attn.o_proj
242# - model.layers.20.self_attn.o_proj
243# - model.layers.76.self_attn.o_proj
244# - model.layers.28.self_attn.o_proj
245# # self_attn.q_proj layers
246# - model.layers.1.self_attn.q_proj
247# - model.layers.6.self_attn.q_proj
248# - model.layers.0.self_attn.q_proj
249# - model.layers.5.self_attn.q_proj
250# - model.layers.2.self_attn.q_proj
251# - model.layers.7.self_attn.q_proj
252# - model.layers.3.self_attn.q_proj
253# - model.layers.4.self_attn.q_proj
254# - model.layers.8.self_attn.q_proj
255# - model.layers.9.self_attn.q_proj
256# - model.layers.61.self_attn.q_proj
257# - model.layers.10.self_attn.q_proj
258# - model.layers.62.self_attn.q_proj
259# - model.layers.36.self_attn.q_proj
260# - model.layers.15.self_attn.q_proj
261# - model.layers.11.self_attn.q_proj
262# - model.layers.17.self_attn.q_proj
263# - model.layers.60.self_attn.q_proj
264# - model.layers.63.self_attn.q_proj
265# - model.layers.64.self_attn.q_proj
266# - model.layers.29.self_attn.q_proj
267# - model.layers.30.self_attn.q_proj
268# - model.layers.55.self_attn.q_proj
269# - model.layers.34.self_attn.q_proj
270# # self_attn.v_proj layers
271# - model.layers.12.self_attn.v_proj
272# - model.layers.16.self_attn.v_proj
273# - model.layers.18.self_attn.v_proj
274# - model.layers.19.self_attn.v_proj
275# - model.layers.20.self_attn.v_proj
276# - model.layers.21.self_attn.v_proj
277# - model.layers.22.self_attn.v_proj
278# - model.layers.23.self_attn.v_proj
279# - model.layers.24.self_attn.v_proj
280# - model.layers.25.self_attn.v_proj
281# - model.layers.26.self_attn.v_proj
282# - model.layers.27.self_attn.v_proj
283# - model.layers.28.self_attn.v_proj
284# - model.layers.29.self_attn.v_proj
285# - model.layers.30.self_attn.v_proj
286# - model.layers.31.self_attn.v_proj
287# - model.layers.32.self_attn.v_proj
288# - model.layers.33.self_attn.v_proj
289# - model.layers.34.self_attn.v_proj
290# - model.layers.35.self_attn.v_proj
291# - model.layers.36.self_attn.v_proj
292# - model.layers.37.self_attn.v_proj
293# - model.layers.38.self_attn.v_proj
294# - model.layers.39.self_attn.v_proj
295
296
297
298sequence_len: 16384
299sample_packing: true
300pad_to_sequence_len: true
301
302# adapter: lora
303# lora_model_dir:
304# lora_r: 32
305# lora_alpha: 16
306# lora_dropout: 0.05
307# lora_target_linear: true
308# lora_fan_in_fan_out:
309
310wandb_project: dolphin-mixtral1x22b
311wandb_entity:
312wandb_watch:
313wandb_name:
314wandb_log_model:
315
316gradient_accumulation_steps: 8
317micro_batch_size: 1
318num_epochs: 3
319optimizer: adamw_8bit
320lr_scheduler: cosine
321learning_rate: 1e-5
322
323train_on_inputs: false
324group_by_length: false
325bf16: auto
326fp16:
327tf32: false
328
329gradient_checkpointing: true
330early_stopping_patience:
331resume_from_checkpoint: /workspace/axolotl2/axolotl/1x22b-out/checkpoint-507
332local_rank:
333logging_steps: 1
334xformers_attention:
335flash_attention: true
336
337warmup_steps: 10
338evals_per_epoch: 4
339eval_table_size:
340eval_max_new_tokens: 128
341saves_per_epoch: 4
342save_total_limit: 2
343debug:
344deepspeed: /workspace/axolotl/deepspeed_configs/zero3_bf16.json
345weight_decay: 0.01
346fsdp:
347fsdp_config:
348special_tokens:
349 eos_token: "<|im_end|>"
350 bos_token: "<s>"
351 # pad_token: "<unk>"
352 unk_token: "<unk>"
353tokens:
354- "<|im_start|>"
355
356
357
358
359| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.9818 | 0.0015 | 1 | 0.9854 |
| 0.4783 | 0.2499 | 169 | 0.5042 |
| 0.464 | 0.4997 | 338 | 0.4755 |
| 0.4561 | 0.7496 | 507 | 0.4593 |
| 0.3981 | 0.9994 | 676 | 0.4553 |
| 0.3725 | 1.2378 | 845 | 0.4525 |
| 0.3624 | 1.4877 | 1014 | 0.4457 |
| 0.359 | 1.7376 | 1183 | 0.4393 |
| 0.375 | 1.9874 | 1352 | 0.4345 |
| 0.2899 | 2.2260 | 1521 | 0.4488 |
| 0.2848 | 2.4759 | 1690 | 0.4473 |
| 0.2935 | 2.7257 | 1859 | 0.4470 |
| 0.2065 | 2.9756 | 2028 | 0.4572 |