Views
No views yet

<|im_start|>system
You are Dolphin, a helpful AI assistant.<|im_end|>
<|im_start|>user
{prompt}<|im_end|>
<|im_start|>assistant

0.4.01base_model: /workspace/axolotl/dbrx-checkpoint
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4trust_remote_code: true
5
6load_in_8bit: false
7# load_in_4bit: true
8strict: false
9
10# adapter: qlora
11# lora_modules_to_save: [embed_tokens, lm_head]
12
13# lora_r: 32
14# lora_alpha: 16
15# lora_dropout: 0.05
16# lora_target_linear: false
17# lora_fan_in_fan_out:
18
19datasets:
20 - path: /workspace/datasets/dolphin-2.9/dolphin201-sharegpt2.jsonl
21 type: sharegpt
22 conversation: chatml
23 # - path: /workspace/datasets/dolphin-2.9/Ultrachat200kunfiltered.jsonl
24 # type: sharegpt
25 # conversation: chatml
26 - path: /workspace/datasets/dolphin-2.9/dolphin-coder-translate-sharegpt2.jsonl
27 type: sharegpt
28 conversation: chatml
29 - path: /workspace/datasets/dolphin-2.9/dolphin-coder-codegen-sharegpt2.jsonl
30 type: sharegpt
31 conversation: chatml
32 - path: /workspace/datasets/dolphin-2.9/m-a-p_Code-Feedback-sharegpt-unfiltered.jsonl
33 type: sharegpt
34 conversation: chatml
35 - path: /workspace/datasets/dolphin-2.9/m-a-p_CodeFeedback-Filtered-Instruction-sharegpt-unfiltered.jsonl
36 type: sharegpt
37 conversation: chatml
38 - path: /workspace/datasets/dolphin-2.9/not_samantha_norefusals.jsonl
39 type: sharegpt
40 conversation: chatml
41 - path: /workspace/datasets/dolphin-2.9/Orca-Math-resort-unfiltered.jsonl
42 type: sharegpt
43 conversation: chatml
44 - path: /workspace/datasets/dolphin-2.9/agent_instruct_react_unfiltered.jsonl
45 type: sharegpt
46 conversation: chatml
47 - path: /workspace/datasets/dolphin-2.9/toolbench_instruct_j1s1_3k_unfiltered.jsonl
48 type: sharegpt
49 conversation: chatml
50 - path: /workspace/datasets/dolphin-2.9/toolbench_negative_unfiltered.jsonl
51 type: sharegpt
52 conversation: chatml
53 - path: /workspace/datasets/dolphin-2.9/toolbench_react_10p_unfiltered.jsonl
54 type: sharegpt
55 conversation: chatml
56 - path: /workspace/datasets/dolphin-2.9/toolbench_tflan_cot_30p_unfiltered.jsonl
57 type: sharegpt
58 conversation: chatml
59 - path: /workspace/datasets/dolphin-2.9/openhermes200k_unfiltered.jsonl
60 type: sharegpt
61 conversation: chatml
62 # - path: /workspace/datasets/dolphin-2.9/SystemConversations.jsonl
63 # type: sharegpt
64 # conversation: chatml
65
66chat_template: chatml
67
68unfrozen_parameters:
69- ^lm_head.weight$
70# ffn.experts.mlp_experts.0.v1 layers
71- transformer.blocks.30.ffn.experts.mlp_experts.0.v1
72- transformer.blocks.32.ffn.experts.mlp_experts.0.v1
73- transformer.blocks.25.ffn.experts.mlp_experts.0.v1
74- transformer.blocks.15.ffn.experts.mlp_experts.0.v1
75- transformer.blocks.22.ffn.experts.mlp_experts.0.v1
76- transformer.blocks.31.ffn.experts.mlp_experts.0.v1
77- transformer.blocks.7.ffn.experts.mlp_experts.0.v1
78- transformer.blocks.21.ffn.experts.mlp_experts.0.v1
79- transformer.blocks.8.ffn.experts.mlp_experts.0.v1
80- transformer.blocks.23.ffn.experts.mlp_experts.0.v1
81# ffn.experts.mlp_experts.0.w1 layers
82- transformer.blocks.7.ffn.experts.mlp_experts.0.w1
83- transformer.blocks.8.ffn.experts.mlp_experts.0.w1
84- transformer.blocks.30.ffn.experts.mlp_experts.0.w1
85- transformer.blocks.4.ffn.experts.mlp_experts.0.w1
86- transformer.blocks.0.ffn.experts.mlp_experts.0.w1
87- transformer.blocks.32.ffn.experts.mlp_experts.0.w1
88- transformer.blocks.6.ffn.experts.mlp_experts.0.w1
89- transformer.blocks.3.ffn.experts.mlp_experts.0.w1
90- transformer.blocks.25.ffn.experts.mlp_experts.0.w1
91- transformer.blocks.5.ffn.experts.mlp_experts.0.w1
92# ffn.experts.mlp_experts.0.w2 layers
93- transformer.blocks.25.ffn.experts.mlp_experts.0.w2
94- transformer.blocks.22.ffn.experts.mlp_experts.0.w2
95- transformer.blocks.27.ffn.experts.mlp_experts.0.w2
96- transformer.blocks.26.ffn.experts.mlp_experts.0.w2
97- transformer.blocks.4.ffn.experts.mlp_experts.0.w2
98- transformer.blocks.29.ffn.experts.mlp_experts.0.w2
99- transformer.blocks.32.ffn.experts.mlp_experts.0.w2
100- transformer.blocks.5.ffn.experts.mlp_experts.0.w2
101- transformer.blocks.7.ffn.experts.mlp_experts.0.w2
102- transformer.blocks.3.ffn.experts.mlp_experts.0.w2
103# ffn.experts.mlp_experts.1.v1 layers
104- transformer.blocks.27.ffn.experts.mlp_experts.1.v1
105- transformer.blocks.25.ffn.experts.mlp_experts.1.v1
106- transformer.blocks.29.ffn.experts.mlp_experts.1.v1
107- transformer.blocks.33.ffn.experts.mlp_experts.1.v1
108- transformer.blocks.23.ffn.experts.mlp_experts.1.v1
109- transformer.blocks.30.ffn.experts.mlp_experts.1.v1
110- transformer.blocks.6.ffn.experts.mlp_experts.1.v1
111- transformer.blocks.21.ffn.experts.mlp_experts.1.v1
112- transformer.blocks.15.ffn.experts.mlp_experts.1.v1
113- transformer.blocks.7.ffn.experts.mlp_experts.1.v1
114# ffn.experts.mlp_experts.1.w1 layers
115- transformer.blocks.0.ffn.experts.mlp_experts.1.w1
116- transformer.blocks.6.ffn.experts.mlp_experts.1.w1
117- transformer.blocks.7.ffn.experts.mlp_experts.1.w1
118- transformer.blocks.4.ffn.experts.mlp_experts.1.w1
119- transformer.blocks.8.ffn.experts.mlp_experts.1.w1
120- transformer.blocks.29.ffn.experts.mlp_experts.1.w1
121- transformer.blocks.33.ffn.experts.mlp_experts.1.w1
122- transformer.blocks.27.ffn.experts.mlp_experts.1.w1
123- transformer.blocks.1.ffn.experts.mlp_experts.1.w1
124- transformer.blocks.10.ffn.experts.mlp_experts.1.w1
125# ffn.experts.mlp_experts.1.w2 layers
126- transformer.blocks.25.ffn.experts.mlp_experts.1.w2
127- transformer.blocks.23.ffn.experts.mlp_experts.1.w2
128- transformer.blocks.27.ffn.experts.mlp_experts.1.w2
129- transformer.blocks.29.ffn.experts.mlp_experts.1.w2
130- transformer.blocks.31.ffn.experts.mlp_experts.1.w2
131- transformer.blocks.4.ffn.experts.mlp_experts.1.w2
132- transformer.blocks.32.ffn.experts.mlp_experts.1.w2
133- transformer.blocks.30.ffn.experts.mlp_experts.1.w2
134- transformer.blocks.21.ffn.experts.mlp_experts.1.w2
135- transformer.blocks.33.ffn.experts.mlp_experts.1.w2
136# ffn.experts.mlp_experts.10.v1 layers
137- transformer.blocks.28.ffn.experts.mlp_experts.10.v1
138- transformer.blocks.34.ffn.experts.mlp_experts.10.v1
139- transformer.blocks.33.ffn.experts.mlp_experts.10.v1
140- transformer.blocks.26.ffn.experts.mlp_experts.10.v1
141- transformer.blocks.32.ffn.experts.mlp_experts.10.v1
142- transformer.blocks.30.ffn.experts.mlp_experts.10.v1
143- transformer.blocks.36.ffn.experts.mlp_experts.10.v1
144- transformer.blocks.24.ffn.experts.mlp_experts.10.v1
145- transformer.blocks.20.ffn.experts.mlp_experts.10.v1
146- transformer.blocks.35.ffn.experts.mlp_experts.10.v1
147# ffn.experts.mlp_experts.10.w1 layers
148- transformer.blocks.24.ffn.experts.mlp_experts.10.w1
149- transformer.blocks.33.ffn.experts.mlp_experts.10.w1
150- transformer.blocks.8.ffn.experts.mlp_experts.10.w1
151- transformer.blocks.7.ffn.experts.mlp_experts.10.w1
152- transformer.blocks.34.ffn.experts.mlp_experts.10.w1
153- transformer.blocks.28.ffn.experts.mlp_experts.10.w1
154- transformer.blocks.30.ffn.experts.mlp_experts.10.w1
155- transformer.blocks.1.ffn.experts.mlp_experts.10.w1
156- transformer.blocks.3.ffn.experts.mlp_experts.10.w1
157- transformer.blocks.5.ffn.experts.mlp_experts.10.w1
158# ffn.experts.mlp_experts.10.w2 layers
159- transformer.blocks.24.ffn.experts.mlp_experts.10.w2
160- transformer.blocks.28.ffn.experts.mlp_experts.10.w2
161- transformer.blocks.23.ffn.experts.mlp_experts.10.w2
162- transformer.blocks.30.ffn.experts.mlp_experts.10.w2
163- transformer.blocks.32.ffn.experts.mlp_experts.10.w2
164- transformer.blocks.3.ffn.experts.mlp_experts.10.w2
165- transformer.blocks.33.ffn.experts.mlp_experts.10.w2
166- transformer.blocks.26.ffn.experts.mlp_experts.10.w2
167- transformer.blocks.2.ffn.experts.mlp_experts.10.w2
168- transformer.blocks.20.ffn.experts.mlp_experts.10.w2
169# ffn.experts.mlp_experts.11.w1 layers
170- transformer.blocks.6.ffn.experts.mlp_experts.11.w1
171- transformer.blocks.8.ffn.experts.mlp_experts.11.w1
172- transformer.blocks.9.ffn.experts.mlp_experts.11.w1
173- transformer.blocks.0.ffn.experts.mlp_experts.11.w1
174- transformer.blocks.10.ffn.experts.mlp_experts.11.w1
175- transformer.blocks.28.ffn.experts.mlp_experts.11.w1
176- transformer.blocks.3.ffn.experts.mlp_experts.11.w1
177- transformer.blocks.5.ffn.experts.mlp_experts.11.w1
178- transformer.blocks.33.ffn.experts.mlp_experts.11.w1
179- transformer.blocks.13.ffn.experts.mlp_experts.11.w1
180# ffn.experts.mlp_experts.11.w2 layers
181- transformer.blocks.27.ffn.experts.mlp_experts.11.w2
182- transformer.blocks.24.ffn.experts.mlp_experts.11.w2
183- transformer.blocks.29.ffn.experts.mlp_experts.11.w2
184- transformer.blocks.30.ffn.experts.mlp_experts.11.w2
185- transformer.blocks.22.ffn.experts.mlp_experts.11.w2
186- transformer.blocks.6.ffn.experts.mlp_experts.11.w2
187- transformer.blocks.25.ffn.experts.mlp_experts.11.w2
188- transformer.blocks.7.ffn.experts.mlp_experts.11.w2
189- transformer.blocks.28.ffn.experts.mlp_experts.11.w2
190- transformer.blocks.5.ffn.experts.mlp_experts.11.w2
191# ffn.experts.mlp_experts.12.v1 layers
192- transformer.blocks.30.ffn.experts.mlp_experts.12.v1
193- transformer.blocks.21.ffn.experts.mlp_experts.12.v1
194- transformer.blocks.27.ffn.experts.mlp_experts.12.v1
195- transformer.blocks.28.ffn.experts.mlp_experts.12.v1
196- transformer.blocks.29.ffn.experts.mlp_experts.12.v1
197- transformer.blocks.8.ffn.experts.mlp_experts.12.v1
198- transformer.blocks.10.ffn.experts.mlp_experts.12.v1
199- transformer.blocks.23.ffn.experts.mlp_experts.12.v1
200- transformer.blocks.6.ffn.experts.mlp_experts.12.v1
201- transformer.blocks.20.ffn.experts.mlp_experts.12.v1
202# ffn.experts.mlp_experts.12.w1 layers
203- transformer.blocks.8.ffn.experts.mlp_experts.12.w1
204- transformer.blocks.1.ffn.experts.mlp_experts.12.w1
205- transformer.blocks.0.ffn.experts.mlp_experts.12.w1
206- transformer.blocks.6.ffn.experts.mlp_experts.12.w1
207- transformer.blocks.9.ffn.experts.mlp_experts.12.w1
208- transformer.blocks.2.ffn.experts.mlp_experts.12.w1
209- transformer.blocks.10.ffn.experts.mlp_experts.12.w1
210- transformer.blocks.17.ffn.experts.mlp_experts.12.w1
211- transformer.blocks.29.ffn.experts.mlp_experts.12.w1
212- transformer.blocks.21.ffn.experts.mlp_experts.12.w1
213# ffn.experts.mlp_experts.12.w2 layers
214- transformer.blocks.6.ffn.experts.mlp_experts.12.w2
215- transformer.blocks.25.ffn.experts.mlp_experts.12.w2
216- transformer.blocks.27.ffn.experts.mlp_experts.12.w2
217- transformer.blocks.8.ffn.experts.mlp_experts.12.w2
218- transformer.blocks.31.ffn.experts.mlp_experts.12.w2
219- transformer.blocks.21.ffn.experts.mlp_experts.12.w2
220- transformer.blocks.2.ffn.experts.mlp_experts.12.w2
221- transformer.blocks.29.ffn.experts.mlp_experts.12.w2
222- transformer.blocks.32.ffn.experts.mlp_experts.12.w2
223- transformer.blocks.30.ffn.experts.mlp_experts.12.w2
224# ffn.experts.mlp_experts.13.v1 layers
225- transformer.blocks.31.ffn.experts.mlp_experts.13.v1
226- transformer.blocks.24.ffn.experts.mlp_experts.13.v1
227- transformer.blocks.30.ffn.experts.mlp_experts.13.v1
228- transformer.blocks.29.ffn.experts.mlp_experts.13.v1
229- transformer.blocks.8.ffn.experts.mlp_experts.13.v1
230- transformer.blocks.10.ffn.experts.mlp_experts.13.v1
231- transformer.blocks.11.ffn.experts.mlp_experts.13.v1
232- transformer.blocks.27.ffn.experts.mlp_experts.13.v1
233- transformer.blocks.25.ffn.experts.mlp_experts.13.v1
234- transformer.blocks.36.ffn.experts.mlp_experts.13.v1
235# ffn.experts.mlp_experts.13.w1 layers
236- transformer.blocks.4.ffn.experts.mlp_experts.13.w1
237- transformer.blocks.10.ffn.experts.mlp_experts.13.w1
238- transformer.blocks.6.ffn.experts.mlp_experts.13.w1
239- transformer.blocks.0.ffn.experts.mlp_experts.13.w1
240- transformer.blocks.3.ffn.experts.mlp_experts.13.w1
241- transformer.blocks.24.ffn.experts.mlp_experts.13.w1
242- transformer.blocks.8.ffn.experts.mlp_experts.13.w1
243- transformer.blocks.1.ffn.experts.mlp_experts.13.w1
244- transformer.blocks.30.ffn.experts.mlp_experts.13.w1
245- transformer.blocks.11.ffn.experts.mlp_experts.13.w1
246# ffn.experts.mlp_experts.13.w2 layers
247- transformer.blocks.24.ffn.experts.mlp_experts.13.w2
248- transformer.blocks.20.ffn.experts.mlp_experts.13.w2
249- transformer.blocks.25.ffn.experts.mlp_experts.13.w2
250- transformer.blocks.27.ffn.experts.mlp_experts.13.w2
251- transformer.blocks.3.ffn.experts.mlp_experts.13.w2
252- transformer.blocks.4.ffn.experts.mlp_experts.13.w2
253- transformer.blocks.29.ffn.experts.mlp_experts.13.w2
254- transformer.blocks.6.ffn.experts.mlp_experts.13.w2
255- transformer.blocks.30.ffn.experts.mlp_experts.13.w2
256- transformer.blocks.31.ffn.experts.mlp_experts.13.w2
257# ffn.experts.mlp_experts.14.v1 layers
258- transformer.blocks.28.ffn.experts.mlp_experts.14.v1
259- transformer.blocks.26.ffn.experts.mlp_experts.14.v1
260- transformer.blocks.29.ffn.experts.mlp_experts.14.v1
261- transformer.blocks.35.ffn.experts.mlp_experts.14.v1
262- transformer.blocks.24.ffn.experts.mlp_experts.14.v1
263- transformer.blocks.8.ffn.experts.mlp_experts.14.v1
264- transformer.blocks.32.ffn.experts.mlp_experts.14.v1
265- transformer.blocks.15.ffn.experts.mlp_experts.14.v1
266- transformer.blocks.11.ffn.experts.mlp_experts.14.v1
267- transformer.blocks.22.ffn.experts.mlp_experts.14.v1
268# ffn.experts.mlp_experts.14.w1 layers
269- transformer.blocks.8.ffn.experts.mlp_experts.14.w1
270- transformer.blocks.4.ffn.experts.mlp_experts.14.w1
271- transformer.blocks.5.ffn.experts.mlp_experts.14.w1
272- transformer.blocks.7.ffn.experts.mlp_experts.14.w1
273- transformer.blocks.3.ffn.experts.mlp_experts.14.w1
274- transformer.blocks.13.ffn.experts.mlp_experts.14.w1
275- transformer.blocks.29.ffn.experts.mlp_experts.14.w1
276- transformer.blocks.6.ffn.experts.mlp_experts.14.w1
277- transformer.blocks.28.ffn.experts.mlp_experts.14.w1
278- transformer.blocks.9.ffn.experts.mlp_experts.14.w1
279# ffn.experts.mlp_experts.14.w2 layers
280- transformer.blocks.26.ffn.experts.mlp_experts.14.w2
281- transformer.blocks.24.ffn.experts.mlp_experts.14.w2
282- transformer.blocks.29.ffn.experts.mlp_experts.14.w2
283- transformer.blocks.28.ffn.experts.mlp_experts.14.w2
284- transformer.blocks.31.ffn.experts.mlp_experts.14.w2
285- transformer.blocks.5.ffn.experts.mlp_experts.14.w2
286- transformer.blocks.4.ffn.experts.mlp_experts.14.w2
287- transformer.blocks.32.ffn.experts.mlp_experts.14.w2
288- transformer.blocks.6.ffn.experts.mlp_experts.14.w2
289- transformer.blocks.22.ffn.experts.mlp_experts.14.w2
290# ffn.experts.mlp_experts.15.v1 layers
291- transformer.blocks.33.ffn.experts.mlp_experts.15.v1
292- transformer.blocks.26.ffn.experts.mlp_experts.15.v1
293- transformer.blocks.31.ffn.experts.mlp_experts.15.v1
294- transformer.blocks.28.ffn.experts.mlp_experts.15.v1
295- transformer.blocks.9.ffn.experts.mlp_experts.15.v1
296- transformer.blocks.34.ffn.experts.mlp_experts.15.v1
297- transformer.blocks.29.ffn.experts.mlp_experts.15.v1
298- transformer.blocks.7.ffn.experts.mlp_experts.15.v1
299- transformer.blocks.17.ffn.experts.mlp_experts.15.v1
300- transformer.blocks.15.ffn.experts.mlp_experts.15.v1
301# ffn.experts.mlp_experts.15.w1 layers
302- transformer.blocks.6.ffn.experts.mlp_experts.15.w1
303- transformer.blocks.9.ffn.experts.mlp_experts.15.w1
304- transformer.blocks.0.ffn.experts.mlp_experts.15.w1
305- transformer.blocks.7.ffn.experts.mlp_experts.15.w1
306- transformer.blocks.14.ffn.experts.mlp_experts.15.w1
307- transformer.blocks.33.ffn.experts.mlp_experts.15.w1
308- transformer.blocks.34.ffn.experts.mlp_experts.15.w1
309- transformer.blocks.10.ffn.experts.mlp_experts.15.w1
310- transformer.blocks.5.ffn.experts.mlp_experts.15.w1
311- transformer.blocks.29.ffn.experts.mlp_experts.15.w1
312# ffn.experts.mlp_experts.15.w2 layers
313- transformer.blocks.28.ffn.experts.mlp_experts.15.w2
314- transformer.blocks.26.ffn.experts.mlp_experts.15.w2
315- transformer.blocks.27.ffn.experts.mlp_experts.15.w2
316- transformer.blocks.29.ffn.experts.mlp_experts.15.w2
317- transformer.blocks.6.ffn.experts.mlp_experts.15.w2
318- transformer.blocks.31.ffn.experts.mlp_experts.15.w2
319- transformer.blocks.7.ffn.experts.mlp_experts.15.w2
320- transformer.blocks.33.ffn.experts.mlp_experts.15.w2
321- transformer.blocks.32.ffn.experts.mlp_experts.15.w2
322- transformer.blocks.25.ffn.experts.mlp_experts.15.w2
323# ffn.experts.mlp_experts.2.v1 layers
324- transformer.blocks.31.ffn.experts.mlp_experts.2.v1
325- transformer.blocks.27.ffn.experts.mlp_experts.2.v1
326- transformer.blocks.28.ffn.experts.mlp_experts.2.v1
327- transformer.blocks.30.ffn.experts.mlp_experts.2.v1
328- transformer.blocks.23.ffn.experts.mlp_experts.2.v1
329- transformer.blocks.32.ffn.experts.mlp_experts.2.v1
330- transformer.blocks.35.ffn.experts.mlp_experts.2.v1
331- transformer.blocks.7.ffn.experts.mlp_experts.2.v1
332- transformer.blocks.21.ffn.experts.mlp_experts.2.v1
333- transformer.blocks.15.ffn.experts.mlp_experts.2.v1
334# ffn.experts.mlp_experts.2.w1 layers
335- transformer.blocks.7.ffn.experts.mlp_experts.2.w1
336- transformer.blocks.6.ffn.experts.mlp_experts.2.w1
337- transformer.blocks.1.ffn.experts.mlp_experts.2.w1
338- transformer.blocks.4.ffn.experts.mlp_experts.2.w1
339- transformer.blocks.5.ffn.experts.mlp_experts.2.w1
340- transformer.blocks.29.ffn.experts.mlp_experts.2.w1
341- transformer.blocks.0.ffn.experts.mlp_experts.2.w1
342- transformer.blocks.9.ffn.experts.mlp_experts.2.w1
343- transformer.blocks.31.ffn.experts.mlp_experts.2.w1
344- transformer.blocks.30.ffn.experts.mlp_experts.2.w1
345# ffn.experts.mlp_experts.2.w2 layers
346- transformer.blocks.26.ffn.experts.mlp_experts.2.w2
347- transformer.blocks.27.ffn.experts.mlp_experts.2.w2
348- transformer.blocks.33.ffn.experts.mlp_experts.2.w2
349- transformer.blocks.5.ffn.experts.mlp_experts.2.w2
350- transformer.blocks.23.ffn.experts.mlp_experts.2.w2
351- transformer.blocks.32.ffn.experts.mlp_experts.2.w2
352- transformer.blocks.28.ffn.experts.mlp_experts.2.w2
353- transformer.blocks.4.ffn.experts.mlp_experts.2.w2
354- transformer.blocks.29.ffn.experts.mlp_experts.2.w2
355- transformer.blocks.30.ffn.experts.mlp_experts.2.w2
356# ffn.experts.mlp_experts.3.v1 layers
357- transformer.blocks.28.ffn.experts.mlp_experts.3.v1
358- transformer.blocks.33.ffn.experts.mlp_experts.3.v1
359- transformer.blocks.36.ffn.experts.mlp_experts.3.v1
360- transformer.blocks.29.ffn.experts.mlp_experts.3.v1
361- transformer.blocks.30.ffn.experts.mlp_experts.3.v1
362- transformer.blocks.7.ffn.experts.mlp_experts.3.v1
363- transformer.blocks.14.ffn.experts.mlp_experts.3.v1
364- transformer.blocks.10.ffn.experts.mlp_experts.3.v1
365- transformer.blocks.31.ffn.experts.mlp_experts.3.v1
366- transformer.blocks.21.ffn.experts.mlp_experts.3.v1
367# ffn.experts.mlp_experts.3.w1 layers
368- transformer.blocks.7.ffn.experts.mlp_experts.3.w1
369- transformer.blocks.0.ffn.experts.mlp_experts.3.w1
370- transformer.blocks.10.ffn.experts.mlp_experts.3.w1
371- transformer.blocks.9.ffn.experts.mlp_experts.3.w1
372- transformer.blocks.29.ffn.experts.mlp_experts.3.w1
373- transformer.blocks.5.ffn.experts.mlp_experts.3.w1
374- transformer.blocks.30.ffn.experts.mlp_experts.3.w1
375- transformer.blocks.4.ffn.experts.mlp_experts.3.w1
376- transformer.blocks.33.ffn.experts.mlp_experts.3.w1
377- transformer.blocks.1.ffn.experts.mlp_experts.3.w1
378# ffn.experts.mlp_experts.3.w2 layers
379- transformer.blocks.28.ffn.experts.mlp_experts.3.w2
380- transformer.blocks.5.ffn.experts.mlp_experts.3.w2
381- transformer.blocks.24.ffn.experts.mlp_experts.3.w2
382- transformer.blocks.31.ffn.experts.mlp_experts.3.w2
383- transformer.blocks.30.ffn.experts.mlp_experts.3.w2
384- transformer.blocks.21.ffn.experts.mlp_experts.3.w2
385- transformer.blocks.32.ffn.experts.mlp_experts.3.w2
386- transformer.blocks.29.ffn.experts.mlp_experts.3.w2
387- transformer.blocks.26.ffn.experts.mlp_experts.3.w2
388- transformer.blocks.2.ffn.experts.mlp_experts.3.w2
389# ffn.experts.mlp_experts.4.v1 layers
390- transformer.blocks.34.ffn.experts.mlp_experts.4.v1
391- transformer.blocks.31.ffn.experts.mlp_experts.4.v1
392- transformer.blocks.26.ffn.experts.mlp_experts.4.v1
393- transformer.blocks.24.ffn.experts.mlp_experts.4.v1
394- transformer.blocks.14.ffn.experts.mlp_experts.4.v1
395- transformer.blocks.32.ffn.experts.mlp_experts.4.v1
396- transformer.blocks.7.ffn.experts.mlp_experts.4.v1
397- transformer.blocks.6.ffn.experts.mlp_experts.4.v1
398- transformer.blocks.20.ffn.experts.mlp_experts.4.v1
399- transformer.blocks.9.ffn.experts.mlp_experts.4.v1
400# ffn.experts.mlp_experts.4.w1 layers
401- transformer.blocks.6.ffn.experts.mlp_experts.4.w1
402- transformer.blocks.4.ffn.experts.mlp_experts.4.w1
403- transformer.blocks.7.ffn.experts.mlp_experts.4.w1
404- transformer.blocks.9.ffn.experts.mlp_experts.4.w1
405- transformer.blocks.0.ffn.experts.mlp_experts.4.w1
406- transformer.blocks.5.ffn.experts.mlp_experts.4.w1
407- transformer.blocks.14.ffn.experts.mlp_experts.4.w1
408- transformer.blocks.34.ffn.experts.mlp_experts.4.w1
409- transformer.blocks.8.ffn.experts.mlp_experts.4.w1
410- transformer.blocks.29.ffn.experts.mlp_experts.4.w1
411# ffn.experts.mlp_experts.4.w2 layers
412- transformer.blocks.25.ffn.experts.mlp_experts.4.w2
413- transformer.blocks.24.ffn.experts.mlp_experts.4.w2
414- transformer.blocks.26.ffn.experts.mlp_experts.4.w2
415- transformer.blocks.5.ffn.experts.mlp_experts.4.w2
416- transformer.blocks.6.ffn.experts.mlp_experts.4.w2
417- transformer.blocks.32.ffn.experts.mlp_experts.4.w2
418- transformer.blocks.4.ffn.experts.mlp_experts.4.w2
419- transformer.blocks.36.ffn.experts.mlp_experts.4.w2
420- transformer.blocks.29.ffn.experts.mlp_experts.4.w2
421- transformer.blocks.27.ffn.experts.mlp_experts.4.w2
422# ffn.experts.mlp_experts.5.v1 layers
423- transformer.blocks.35.ffn.experts.mlp_experts.5.v1
424- transformer.blocks.30.ffn.experts.mlp_experts.5.v1
425- transformer.blocks.28.ffn.experts.mlp_experts.5.v1
426- transformer.blocks.32.ffn.experts.mlp_experts.5.v1
427- transformer.blocks.27.ffn.experts.mlp_experts.5.v1
428- transformer.blocks.26.ffn.experts.mlp_experts.5.v1
429- transformer.blocks.33.ffn.experts.mlp_experts.5.v1
430- transformer.blocks.29.ffn.experts.mlp_experts.5.v1
431- transformer.blocks.8.ffn.experts.mlp_experts.5.v1
432- transformer.blocks.7.ffn.experts.mlp_experts.5.v1
433# ffn.experts.mlp_experts.5.w1 layers
434- transformer.blocks.0.ffn.experts.mlp_experts.5.w1
435- transformer.blocks.6.ffn.experts.mlp_experts.5.w1
436- transformer.blocks.7.ffn.experts.mlp_experts.5.w1
437- transformer.blocks.9.ffn.experts.mlp_experts.5.w1
438- transformer.blocks.8.ffn.experts.mlp_experts.5.w1
439- transformer.blocks.12.ffn.experts.mlp_experts.5.w1
440- transformer.blocks.3.ffn.experts.mlp_experts.5.w1
441- transformer.blocks.5.ffn.experts.mlp_experts.5.w1
442- transformer.blocks.4.ffn.experts.mlp_experts.5.w1
443- transformer.blocks.33.ffn.experts.mlp_experts.5.w1
444# ffn.experts.mlp_experts.5.w2 layers
445- transformer.blocks.26.ffn.experts.mlp_experts.5.w2
446- transformer.blocks.28.ffn.experts.mlp_experts.5.w2
447- transformer.blocks.6.ffn.experts.mlp_experts.5.w2
448- transformer.blocks.33.ffn.experts.mlp_experts.5.w2
449- transformer.blocks.5.ffn.experts.mlp_experts.5.w2
450- transformer.blocks.27.ffn.experts.mlp_experts.5.w2
451- transformer.blocks.3.ffn.experts.mlp_experts.5.w2
452- transformer.blocks.29.ffn.experts.mlp_experts.5.w2
453- transformer.blocks.25.ffn.experts.mlp_experts.5.w2
454- transformer.blocks.7.ffn.experts.mlp_experts.5.w2
455# ffn.experts.mlp_experts.6.v1 layers
456- transformer.blocks.34.ffn.experts.mlp_experts.6.v1
457- transformer.blocks.31.ffn.experts.mlp_experts.6.v1
458- transformer.blocks.30.ffn.experts.mlp_experts.6.v1
459- transformer.blocks.26.ffn.experts.mlp_experts.6.v1
460- transformer.blocks.35.ffn.experts.mlp_experts.6.v1
461- transformer.blocks.20.ffn.experts.mlp_experts.6.v1
462- transformer.blocks.15.ffn.experts.mlp_experts.6.v1
463- transformer.blocks.29.ffn.experts.mlp_experts.6.v1
464- transformer.blocks.10.ffn.experts.mlp_experts.6.v1
465- transformer.blocks.24.ffn.experts.mlp_experts.6.v1
466# ffn.experts.mlp_experts.6.w1 layers
467- transformer.blocks.0.ffn.experts.mlp_experts.6.w1
468- transformer.blocks.10.ffn.experts.mlp_experts.6.w1
469- transformer.blocks.9.ffn.experts.mlp_experts.6.w1
470- transformer.blocks.30.ffn.experts.mlp_experts.6.w1
471- transformer.blocks.4.ffn.experts.mlp_experts.6.w1
472- transformer.blocks.34.ffn.experts.mlp_experts.6.w1
473- transformer.blocks.26.ffn.experts.mlp_experts.6.w1
474- transformer.blocks.2.ffn.experts.mlp_experts.6.w1
475- transformer.blocks.29.ffn.experts.mlp_experts.6.w1
476- transformer.blocks.8.ffn.experts.mlp_experts.6.w1
477# ffn.experts.mlp_experts.6.w2 layers
478- transformer.blocks.24.ffn.experts.mlp_experts.6.w2
479- transformer.blocks.26.ffn.experts.mlp_experts.6.w2
480- transformer.blocks.32.ffn.experts.mlp_experts.6.w2
481- transformer.blocks.30.ffn.experts.mlp_experts.6.w2
482- transformer.blocks.25.ffn.experts.mlp_experts.6.w2
483- transformer.blocks.31.ffn.experts.mlp_experts.6.w2
484- transformer.blocks.20.ffn.experts.mlp_experts.6.w2
485- transformer.blocks.4.ffn.experts.mlp_experts.6.w2
486- transformer.blocks.2.ffn.experts.mlp_experts.6.w2
487- transformer.blocks.9.ffn.experts.mlp_experts.6.w2
488# ffn.experts.mlp_experts.7.v1 layers
489- transformer.blocks.27.ffn.experts.mlp_experts.7.v1
490- transformer.blocks.28.ffn.experts.mlp_experts.7.v1
491- transformer.blocks.33.ffn.experts.mlp_experts.7.v1
492- transformer.blocks.29.ffn.experts.mlp_experts.7.v1
493- transformer.blocks.24.ffn.experts.mlp_experts.7.v1
494- transformer.blocks.11.ffn.experts.mlp_experts.7.v1
495- transformer.blocks.12.ffn.experts.mlp_experts.7.v1
496- transformer.blocks.10.ffn.experts.mlp_experts.7.v1
497- transformer.blocks.23.ffn.experts.mlp_experts.7.v1
498- transformer.blocks.34.ffn.experts.mlp_experts.7.v1
499# ffn.experts.mlp_experts.7.w1 layers
500- transformer.blocks.12.ffn.experts.mlp_experts.7.w1
501- transformer.blocks.0.ffn.experts.mlp_experts.7.w1
502- transformer.blocks.5.ffn.experts.mlp_experts.7.w1
503- transformer.blocks.29.ffn.experts.mlp_experts.7.w1
504- transformer.blocks.10.ffn.experts.mlp_experts.7.w1
505- transformer.blocks.4.ffn.experts.mlp_experts.7.w1
506- transformer.blocks.3.ffn.experts.mlp_experts.7.w1
507- transformer.blocks.8.ffn.experts.mlp_experts.7.w1
508- transformer.blocks.34.ffn.experts.mlp_experts.7.w1
509- transformer.blocks.33.ffn.experts.mlp_experts.7.w1
510# ffn.experts.mlp_experts.7.w2 layers
511- transformer.blocks.23.ffn.experts.mlp_experts.7.w2
512- transformer.blocks.24.ffn.experts.mlp_experts.7.w2
513- transformer.blocks.31.ffn.experts.mlp_experts.7.w2
514- transformer.blocks.28.ffn.experts.mlp_experts.7.w2
515- transformer.blocks.27.ffn.experts.mlp_experts.7.w2
516- transformer.blocks.5.ffn.experts.mlp_experts.7.w2
517- transformer.blocks.25.ffn.experts.mlp_experts.7.w2
518- transformer.blocks.29.ffn.experts.mlp_experts.7.w2
519- transformer.blocks.3.ffn.experts.mlp_experts.7.w2
520- transformer.blocks.33.ffn.experts.mlp_experts.7.w2
521# ffn.experts.mlp_experts.8.v1 layers
522- transformer.blocks.30.ffn.experts.mlp_experts.8.v1
523- transformer.blocks.27.ffn.experts.mlp_experts.8.v1
524- transformer.blocks.20.ffn.experts.mlp_experts.8.v1
525- transformer.blocks.32.ffn.experts.mlp_experts.8.v1
526- transformer.blocks.34.ffn.experts.mlp_experts.8.v1
527- transformer.blocks.33.ffn.experts.mlp_experts.8.v1
528- transformer.blocks.9.ffn.experts.mlp_experts.8.v1
529- transformer.blocks.7.ffn.experts.mlp_experts.8.v1
530- transformer.blocks.6.ffn.experts.mlp_experts.8.v1
531- transformer.blocks.24.ffn.experts.mlp_experts.8.v1
532# ffn.experts.mlp_experts.8.w1 layers
533- transformer.blocks.7.ffn.experts.mlp_experts.8.w1
534- transformer.blocks.6.ffn.experts.mlp_experts.8.w1
535- transformer.blocks.0.ffn.experts.mlp_experts.8.w1
536- transformer.blocks.9.ffn.experts.mlp_experts.8.w1
537- transformer.blocks.3.ffn.experts.mlp_experts.8.w1
538- transformer.blocks.2.ffn.experts.mlp_experts.8.w1
539- transformer.blocks.8.ffn.experts.mlp_experts.8.w1
540- transformer.blocks.30.ffn.experts.mlp_experts.8.w1
541- transformer.blocks.24.ffn.experts.mlp_experts.8.w1
542- transformer.blocks.1.ffn.experts.mlp_experts.8.w1
543# ffn.experts.mlp_experts.8.w2 layers
544- transformer.blocks.32.ffn.experts.mlp_experts.8.w2
545- transformer.blocks.24.ffn.experts.mlp_experts.8.w2
546- transformer.blocks.27.ffn.experts.mlp_experts.8.w2
547- transformer.blocks.30.ffn.experts.mlp_experts.8.w2
548- transformer.blocks.31.ffn.experts.mlp_experts.8.w2
549- transformer.blocks.28.ffn.experts.mlp_experts.8.w2
550- transformer.blocks.2.ffn.experts.mlp_experts.8.w2
551- transformer.blocks.3.ffn.experts.mlp_experts.8.w2
552- transformer.blocks.23.ffn.experts.mlp_experts.8.w2
553- transformer.blocks.29.ffn.experts.mlp_experts.8.w2
554# ffn.experts.mlp_experts.9.v1 layers
555- transformer.blocks.31.ffn.experts.mlp_experts.9.v1
556- transformer.blocks.27.ffn.experts.mlp_experts.9.v1
557- transformer.blocks.29.ffn.experts.mlp_experts.9.v1
558- transformer.blocks.33.ffn.experts.mlp_experts.9.v1
559- transformer.blocks.25.ffn.experts.mlp_experts.9.v1
560- transformer.blocks.14.ffn.experts.mlp_experts.9.v1
561- transformer.blocks.32.ffn.experts.mlp_experts.9.v1
562- transformer.blocks.7.ffn.experts.mlp_experts.9.v1
563- transformer.blocks.9.ffn.experts.mlp_experts.9.v1
564- transformer.blocks.34.ffn.experts.mlp_experts.9.v1
565# ffn.experts.mlp_experts.9.w1 layers
566- transformer.blocks.7.ffn.experts.mlp_experts.9.w1
567- transformer.blocks.1.ffn.experts.mlp_experts.9.w1
568- transformer.blocks.9.ffn.experts.mlp_experts.9.w1
569- transformer.blocks.2.ffn.experts.mlp_experts.9.w1
570- transformer.blocks.27.ffn.experts.mlp_experts.9.w1
571- transformer.blocks.12.ffn.experts.mlp_experts.9.w1
572- transformer.blocks.4.ffn.experts.mlp_experts.9.w1
573- transformer.blocks.6.ffn.experts.mlp_experts.9.w1
574- transformer.blocks.19.ffn.experts.mlp_experts.9.w1
575- transformer.blocks.8.ffn.experts.mlp_experts.9.w1
576# ffn.experts.mlp_experts.9.w2 layers
577- transformer.blocks.26.ffn.experts.mlp_experts.9.w2
578- transformer.blocks.25.ffn.experts.mlp_experts.9.w2
579- transformer.blocks.28.ffn.experts.mlp_experts.9.w2
580- transformer.blocks.27.ffn.experts.mlp_experts.9.w2
581- transformer.blocks.31.ffn.experts.mlp_experts.9.w2
582- transformer.blocks.29.ffn.experts.mlp_experts.9.w2
583- transformer.blocks.7.ffn.experts.mlp_experts.9.w2
584- transformer.blocks.34.ffn.experts.mlp_experts.9.w2
585- transformer.blocks.2.ffn.experts.mlp_experts.9.w2
586- transformer.blocks.33.ffn.experts.mlp_experts.9.w2
587# ffn.router.layer layers
588- transformer.blocks.2.ffn.router.layer
589- transformer.blocks.3.ffn.router.layer
590- transformer.blocks.4.ffn.router.layer
591- transformer.blocks.5.ffn.router.layer
592- transformer.blocks.6.ffn.router.layer
593- transformer.blocks.7.ffn.router.layer
594- transformer.blocks.8.ffn.router.layer
595- transformer.blocks.9.ffn.router.layer
596- transformer.blocks.10.ffn.router.layer
597- transformer.blocks.11.ffn.router.layer
598# norm_attn_norm.attn.Wqkv layers
599- transformer.blocks.16.norm_attn_norm.attn.Wqkv
600- transformer.blocks.15.norm_attn_norm.attn.Wqkv
601- transformer.blocks.11.norm_attn_norm.attn.Wqkv
602- transformer.blocks.14.norm_attn_norm.attn.Wqkv
603- transformer.blocks.12.norm_attn_norm.attn.Wqkv
604- transformer.blocks.20.norm_attn_norm.attn.Wqkv
605- transformer.blocks.10.norm_attn_norm.attn.Wqkv
606- transformer.blocks.9.norm_attn_norm.attn.Wqkv
607- transformer.blocks.19.norm_attn_norm.attn.Wqkv
608- transformer.blocks.18.norm_attn_norm.attn.Wqkv
609# norm_attn_norm.attn.out_proj layers
610- transformer.blocks.1.norm_attn_norm.attn.out_proj
611- transformer.blocks.18.norm_attn_norm.attn.out_proj
612- transformer.blocks.2.norm_attn_norm.attn.out_proj
613- transformer.blocks.16.norm_attn_norm.attn.out_proj
614- transformer.blocks.0.norm_attn_norm.attn.out_proj
615- transformer.blocks.39.norm_attn_norm.attn.out_proj
616- transformer.blocks.23.norm_attn_norm.attn.out_proj
617- transformer.blocks.8.norm_attn_norm.attn.out_proj
618- transformer.blocks.24.norm_attn_norm.attn.out_proj
619- transformer.blocks.19.norm_attn_norm.attn.out_proj
620# norm_attn_norm.norm_1 layers
621- transformer.blocks.0.norm_attn_norm.norm_1
622- transformer.blocks.1.norm_attn_norm.norm_1
623- transformer.blocks.2.norm_attn_norm.norm_1
624- transformer.blocks.3.norm_attn_norm.norm_1
625- transformer.blocks.4.norm_attn_norm.norm_1
626- transformer.blocks.5.norm_attn_norm.norm_1
627- transformer.blocks.6.norm_attn_norm.norm_1
628- transformer.blocks.7.norm_attn_norm.norm_1
629- transformer.blocks.8.norm_attn_norm.norm_1
630- transformer.blocks.9.norm_attn_norm.norm_1
631# norm_attn_norm.norm_2 layers
632- transformer.blocks.0.norm_attn_norm.norm_2
633- transformer.blocks.1.norm_attn_norm.norm_2
634- transformer.blocks.2.norm_attn_norm.norm_2
635- transformer.blocks.3.norm_attn_norm.norm_2
636- transformer.blocks.4.norm_attn_norm.norm_2
637- transformer.blocks.5.norm_attn_norm.norm_2
638- transformer.blocks.6.norm_attn_norm.norm_2
639- transformer.blocks.7.norm_attn_norm.norm_2
640- transformer.blocks.8.norm_attn_norm.norm_2
641- transformer.blocks.9.norm_attn_norm.norm_2
642# transformer.norm_f layers
643# transformer.wte layers
644# ffn.experts.mlp_experts.11.v1 layers
645- transformer.blocks.29.ffn.experts.mlp_experts.11.v1
646- transformer.blocks.27.ffn.experts.mlp_experts.11.v1
647- transformer.blocks.30.ffn.experts.mlp_experts.11.v1
648- transformer.blocks.28.ffn.experts.mlp_experts.11.v1
649- transformer.blocks.22.ffn.experts.mlp_experts.11.v1
650- transformer.blocks.7.ffn.experts.mlp_experts.11.v1
651- transformer.blocks.24.ffn.experts.mlp_experts.11.v1
652- transformer.blocks.8.ffn.experts.mlp_experts.11.v1
653- transformer.blocks.6.ffn.experts.mlp_experts.11.v1
654- transformer.blocks.12.ffn.experts.mlp_experts.11.v1
655
656
657
658dataset_prepared_path: dbrx2
659val_set_size: 0.01
660output_dir: ./out
661
662sequence_len: 4096
663sample_packing: true
664pad_to_sequence_len: true
665
666wandb_project: dolphin-2.9-Dbrx
667wandb_watch:
668wandb_run_id:
669wandb_log_model:
670
671gradient_accumulation_steps: 8
672micro_batch_size: 1
673num_epochs: 1
674optimizer: paged_adamw_8bit
675lr_scheduler: cosine
676learning_rate: 1e-5
677
678train_on_inputs: false
679group_by_length: false
680bf16: auto
681fp16:
682tf32: true
683
684gradient_checkpointing: true
685gradient_checkpointing_kwargs:
686 use_reentrant: false
687early_stopping_patience:
688# resume_from_checkpoint: /workspace/axolotl/dbrx-checkpoint
689logging_steps: 1
690xformers_attention:
691flash_attention: true
692
693warmup_steps: 10
694evals_per_epoch: 4
695eval_table_size:
696saves_per_epoch: 4
697save_total_limit: 2
698save_steps:
699debug:
700deepspeed: /workspace/axolotl/deepspeed_configs/zero3_bf16_cpuoffload_params.json
701weight_decay: 0.05
702fsdp:
703fsdp_config:
704special_tokens:
705 bos_token: "<|endoftext|>"
706 eos_token: "<|im_end|>"
707 pad_token: "<|pad|>"
708 unk_token: "<|endoftext|>"
709tokens:
710 - "<|im_start|>"
711 - "<|im_end|>"
712
713| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.4009 | 0.0 | 1 | 0.4328 |
| 0.413 | 0.25 | 587 | 0.4408 |
| 0.3626 | 0.5 | 1174 | 0.4368 |
| 0.3896 | 0.75 | 1761 | 0.4336 |