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: 01-ai/Yi-1.5-34B-32k
2model_type: LlamaForCausalLM
3tokenizer_type: LlamaTokenizer
4trust_remote_code: true
5
6# load_in_8bit: false
7load_in_4bit: true
8# strict: false
9
10adapter: qlora
11lora_modules_to_save: [embed_tokens, lm_head]
12
13lora_r: 32
14lora_alpha: 16
15lora_dropout: 0.05
16lora_target_linear: false
17lora_fan_in_fan_out:
18
19datasets:
20 - path: /workspace/datasets/dolphin-2.9.3/dolphin201-sharegpt2.jsonl
21 type: sharegpt
22 conversation: chatml
23 - path: /workspace/datasets/dolphin-2.9.3/SystemChat_filtered_sharegpt.jsonl
24 type: sharegpt
25 conversation: chatml
26 - path: /workspace/datasets/dolphin-2.9.3/SystemChat_multilingual_sharegpt.jsonl
27 type: sharegpt
28 conversation: chatml
29 - path: /workspace/datasets/dolphin-2.9.3/dolphin-coder-translate-sharegpt2.jsonl
30 type: sharegpt
31 conversation: chatml
32 - path: /workspace/datasets/dolphin-2.9.3/dolphin-coder-codegen-sharegpt2.jsonl
33 type: sharegpt
34 conversation: chatml
35 - path: /workspace/datasets/dolphin-2.9.3/m-a-p_Code-Feedback-sharegpt-unfiltered.jsonl
36 type: sharegpt
37 conversation: chatml
38 - path: /workspace/datasets/dolphin-2.9.3/m-a-p_CodeFeedback-Filtered-Instruction-sharegpt-unfiltered.jsonl
39 type: sharegpt
40 conversation: chatml
41 - path: /workspace/datasets/dolphin-2.9.3/not_samantha_norefusals.jsonl
42 type: sharegpt
43 conversation: chatml
44 - path: /workspace/datasets/dolphin-2.9.3/Orca-Math-resort-unfiltered.jsonl
45 type: sharegpt
46 conversation: chatml
47 - path: /workspace/datasets/dolphin-2.9.3/agent_instruct_react_unfiltered.jsonl
48 type: sharegpt
49 conversation: chatml
50 - path: /workspace/datasets/dolphin-2.9.3/toolbench_instruct_j1s1_3k_unfiltered.jsonl
51 type: sharegpt
52 conversation: chatml
53 - path: /workspace/datasets/dolphin-2.9.3/toolbench_negative_unfiltered.jsonl
54 type: sharegpt
55 conversation: chatml
56 - path: /workspace/datasets/dolphin-2.9.3/toolbench_react_10p_unfiltered.jsonl
57 type: sharegpt
58 conversation: chatml
59 - path: /workspace/datasets/dolphin-2.9.3/toolbench_tflan_cot_30p_unfiltered.jsonl
60 type: sharegpt
61 conversation: chatml
62 - path: /workspace/datasets/dolphin-2.9.3/openhermes200k_unfiltered.jsonl
63 type: sharegpt
64 conversation: chatml
65
66chat_template: chatml
67
68dataset_prepared_path: dolphin-2.9.3-yi34b-prepared
69val_set_size: 0.01
70output_dir: ./dolphin-2.9.3-out
71
72sequence_len: 8192
73sample_packing: true
74pad_to_sequence_len: true
75
76wandb_project: dolphin-2.9.3-yi-1.5-34b
77wandb_watch:
78wandb_run_id:
79wandb_log_model:
80
81gradient_accumulation_steps: 8
82micro_batch_size: 1
83num_epochs: 3
84optimizer: adamw_8bit
85lr_scheduler: cosine
86learning_rate: 1e-5
87
88train_on_inputs: false
89group_by_length: false
90bf16: auto
91fp16:
92tf32: false
93
94gradient_checkpointing: true
95gradient_checkpointing_kwargs:
96 use_reentrant: false
97early_stopping_patience:
98logging_steps: 1
99xformers_attention:
100flash_attention: true
101
102warmup_steps: 10
103# evals_per_epoch: 4
104eval_table_size:
105saves_per_epoch: 4
106save_total_limit: 2
107save_steps:
108debug:
109deepspeed: /workspace/axolotl/deepspeed_configs/zero3_bf16.json
110weight_decay: 0.05
111fsdp:
112fsdp_config:
113special_tokens:
114 bos_token: "<|startoftext|>"
115 eos_token: "<|im_end|>"
116 pad_token: "<unk>"
117 unk_token: "<unk>"
118tokens:
119 - "<|im_start|>"
120
121#unfrozen_parameters:
122lora_target_modules:
123 # input_layernorm layers
124 # - model.layers.0.input_layernorm
125 # - model.layers.1.input_layernorm
126 # - model.layers.2.input_layernorm
127 # - model.layers.3.input_layernorm
128 # - model.layers.4.input_layernorm
129 # - model.layers.5.input_layernorm
130 # - model.layers.6.input_layernorm
131 # - model.layers.7.input_layernorm
132 # - model.layers.8.input_layernorm
133 # - model.layers.9.input_layernorm
134 # - model.layers.10.input_layernorm
135 # - model.layers.11.input_layernorm
136 # - model.layers.12.input_layernorm
137 # - model.layers.13.input_layernorm
138 # - model.layers.14.input_layernorm
139 # - model.layers.15.input_layernorm
140 # - model.layers.16.input_layernorm
141 # - model.layers.17.input_layernorm
142 # - model.layers.18.input_layernorm
143 # - model.layers.19.input_layernorm
144 # - model.layers.20.input_layernorm
145 # - model.layers.21.input_layernorm
146 # - model.layers.22.input_layernorm
147 # - model.layers.23.input_layernorm
148 # - model.layers.24.input_layernorm
149 # - model.layers.25.input_layernorm
150 # - model.layers.26.input_layernorm
151 # - model.layers.27.input_layernorm
152 # - model.layers.28.input_layernorm
153 # - model.layers.29.input_layernorm
154 - lm_head
155 # mlp.down_proj layers
156 - model.layers.44.mlp.down_proj
157 - model.layers.45.mlp.down_proj
158 - model.layers.46.mlp.down_proj
159 - model.layers.47.mlp.down_proj
160 - model.layers.43.mlp.down_proj
161 - model.layers.48.mlp.down_proj
162 - model.layers.49.mlp.down_proj
163 - model.layers.42.mlp.down_proj
164 - model.layers.50.mlp.down_proj
165 - model.layers.41.mlp.down_proj
166 - model.layers.51.mlp.down_proj
167 - model.layers.52.mlp.down_proj
168 - model.layers.39.mlp.down_proj
169 - model.layers.40.mlp.down_proj
170 - model.layers.53.mlp.down_proj
171 - model.layers.54.mlp.down_proj
172 - model.layers.38.mlp.down_proj
173 - model.layers.56.mlp.down_proj
174 - model.layers.55.mlp.down_proj
175 - model.layers.37.mlp.down_proj
176 - model.layers.36.mlp.down_proj
177 - model.layers.57.mlp.down_proj
178 - model.layers.35.mlp.down_proj
179 - model.layers.12.mlp.down_proj
180 - model.layers.13.mlp.down_proj
181 - model.layers.16.mlp.down_proj
182 - model.layers.14.mlp.down_proj
183 - model.layers.11.mlp.down_proj
184 - model.layers.34.mlp.down_proj
185 - model.layers.17.mlp.down_proj
186 # mlp.gate_proj layers
187 - model.layers.57.mlp.gate_proj
188 - model.layers.58.mlp.gate_proj
189 - model.layers.56.mlp.gate_proj
190 - model.layers.55.mlp.gate_proj
191 - model.layers.54.mlp.gate_proj
192 - model.layers.35.mlp.gate_proj
193 - model.layers.34.mlp.gate_proj
194 - model.layers.53.mlp.gate_proj
195 - model.layers.26.mlp.gate_proj
196 - model.layers.52.mlp.gate_proj
197 - model.layers.25.mlp.gate_proj
198 - model.layers.33.mlp.gate_proj
199 - model.layers.51.mlp.gate_proj
200 - model.layers.18.mlp.gate_proj
201 - model.layers.32.mlp.gate_proj
202 - model.layers.36.mlp.gate_proj
203 - model.layers.24.mlp.gate_proj
204 - model.layers.17.mlp.gate_proj
205 - model.layers.23.mlp.gate_proj
206 - model.layers.31.mlp.gate_proj
207 - model.layers.50.mlp.gate_proj
208 - model.layers.19.mlp.gate_proj
209 - model.layers.15.mlp.gate_proj
210 - model.layers.27.mlp.gate_proj
211 - model.layers.37.mlp.gate_proj
212 - model.layers.14.mlp.gate_proj
213 - model.layers.39.mlp.gate_proj
214 - model.layers.11.mlp.gate_proj
215 - model.layers.29.mlp.gate_proj
216 - model.layers.28.mlp.gate_proj
217 # mlp.up_proj layers
218 - model.layers.21.mlp.up_proj
219 - model.layers.48.mlp.up_proj
220 - model.layers.49.mlp.up_proj
221 - model.layers.24.mlp.up_proj
222 - model.layers.47.mlp.up_proj
223 - model.layers.25.mlp.up_proj
224 - model.layers.23.mlp.up_proj
225 - model.layers.50.mlp.up_proj
226 - model.layers.14.mlp.up_proj
227 - model.layers.46.mlp.up_proj
228 - model.layers.26.mlp.up_proj
229 - model.layers.27.mlp.up_proj
230 - model.layers.20.mlp.up_proj
231 - model.layers.13.mlp.up_proj
232 - model.layers.51.mlp.up_proj
233 - model.layers.28.mlp.up_proj
234 - model.layers.45.mlp.up_proj
235 - model.layers.22.mlp.up_proj
236 - model.layers.52.mlp.up_proj
237 - model.layers.12.mlp.up_proj
238 - model.layers.29.mlp.up_proj
239 - model.layers.44.mlp.up_proj
240 - model.layers.53.mlp.up_proj
241 - model.layers.11.mlp.up_proj
242 - model.layers.42.mlp.up_proj
243 - model.layers.30.mlp.up_proj
244 - model.layers.43.mlp.up_proj
245 - model.layers.19.mlp.up_proj
246 - model.layers.54.mlp.up_proj
247 - model.layers.40.mlp.up_proj
248 - model.embed_tokens
249 # model.norm layers
250 # post_attention_layernorm layers
251 # - model.layers.0.post_attention_layernorm
252 # - model.layers.1.post_attention_layernorm
253 # - model.layers.2.post_attention_layernorm
254 # - model.layers.3.post_attention_layernorm
255 # - model.layers.4.post_attention_layernorm
256 # - model.layers.5.post_attention_layernorm
257 # - model.layers.6.post_attention_layernorm
258 # - model.layers.7.post_attention_layernorm
259 # - model.layers.8.post_attention_layernorm
260 # - model.layers.9.post_attention_layernorm
261 # - model.layers.10.post_attention_layernorm
262 # - model.layers.11.post_attention_layernorm
263 # - model.layers.12.post_attention_layernorm
264 # - model.layers.13.post_attention_layernorm
265 # - model.layers.14.post_attention_layernorm
266 # - model.layers.15.post_attention_layernorm
267 # - model.layers.16.post_attention_layernorm
268 # - model.layers.17.post_attention_layernorm
269 # - model.layers.18.post_attention_layernorm
270 # - model.layers.19.post_attention_layernorm
271 # - model.layers.20.post_attention_layernorm
272 # - model.layers.21.post_attention_layernorm
273 # - model.layers.22.post_attention_layernorm
274 # - model.layers.23.post_attention_layernorm
275 # - model.layers.24.post_attention_layernorm
276 # - model.layers.25.post_attention_layernorm
277 # - model.layers.26.post_attention_layernorm
278 # - model.layers.27.post_attention_layernorm
279 # - model.layers.28.post_attention_layernorm
280 # - model.layers.29.post_attention_layernorm
281 # self_attn.k_proj layers
282 - model.layers.55.self_attn.k_proj
283 - model.layers.51.self_attn.k_proj
284 - model.layers.53.self_attn.k_proj
285 - model.layers.56.self_attn.k_proj
286 - model.layers.54.self_attn.k_proj
287 - model.layers.57.self_attn.k_proj
288 - model.layers.52.self_attn.k_proj
289 - model.layers.59.self_attn.k_proj
290 - model.layers.49.self_attn.k_proj
291 - model.layers.48.self_attn.k_proj
292 - model.layers.47.self_attn.k_proj
293 - model.layers.41.self_attn.k_proj
294 - model.layers.58.self_attn.k_proj
295 - model.layers.40.self_attn.k_proj
296 - model.layers.46.self_attn.k_proj
297 - model.layers.44.self_attn.k_proj
298 - model.layers.50.self_attn.k_proj
299 - model.layers.43.self_attn.k_proj
300 - model.layers.39.self_attn.k_proj
301 - model.layers.42.self_attn.k_proj
302 - model.layers.45.self_attn.k_proj
303 - model.layers.33.self_attn.k_proj
304 - model.layers.37.self_attn.k_proj
305 - model.layers.17.self_attn.k_proj
306 - model.layers.24.self_attn.k_proj
307 - model.layers.21.self_attn.k_proj
308 - model.layers.25.self_attn.k_proj
309 - model.layers.23.self_attn.k_proj
310 - model.layers.35.self_attn.k_proj
311 - model.layers.20.self_attn.k_proj
312 # self_attn.o_proj layers
313 - model.layers.53.self_attn.o_proj
314 - model.layers.55.self_attn.o_proj
315 - model.layers.54.self_attn.o_proj
316 - model.layers.42.self_attn.o_proj
317 - model.layers.52.self_attn.o_proj
318 - model.layers.51.self_attn.o_proj
319 - model.layers.50.self_attn.o_proj
320 - model.layers.1.self_attn.o_proj
321 - model.layers.40.self_attn.o_proj
322 - model.layers.37.self_attn.o_proj
323 - model.layers.34.self_attn.o_proj
324 - model.layers.36.self_attn.o_proj
325 - model.layers.41.self_attn.o_proj
326 - model.layers.35.self_attn.o_proj
327 - model.layers.46.self_attn.o_proj
328 - model.layers.27.self_attn.o_proj
329 - model.layers.33.self_attn.o_proj
330 - model.layers.30.self_attn.o_proj
331 - model.layers.43.self_attn.o_proj
332 - model.layers.39.self_attn.o_proj
333 - model.layers.17.self_attn.o_proj
334 - model.layers.28.self_attn.o_proj
335 - model.layers.48.self_attn.o_proj
336 - model.layers.31.self_attn.o_proj
337 - model.layers.29.self_attn.o_proj
338 - model.layers.38.self_attn.o_proj
339 - model.layers.47.self_attn.o_proj
340 - model.layers.56.self_attn.o_proj
341 - model.layers.32.self_attn.o_proj
342 - model.layers.4.self_attn.o_proj
343 # self_attn.q_proj layers
344 - model.layers.1.self_attn.q_proj
345 - model.layers.3.self_attn.q_proj
346 - model.layers.4.self_attn.q_proj
347 - model.layers.5.self_attn.q_proj
348 - model.layers.2.self_attn.q_proj
349 - model.layers.0.self_attn.q_proj
350 - model.layers.6.self_attn.q_proj
351 - model.layers.8.self_attn.q_proj
352 - model.layers.7.self_attn.q_proj
353 - model.layers.10.self_attn.q_proj
354 - model.layers.36.self_attn.q_proj
355 - model.layers.11.self_attn.q_proj
356 - model.layers.9.self_attn.q_proj
357 - model.layers.35.self_attn.q_proj
358 - model.layers.28.self_attn.q_proj
359 - model.layers.34.self_attn.q_proj
360 - model.layers.27.self_attn.q_proj
361 - model.layers.14.self_attn.q_proj
362 - model.layers.29.self_attn.q_proj
363 - model.layers.12.self_attn.q_proj
364 - model.layers.33.self_attn.q_proj
365 - model.layers.30.self_attn.q_proj
366 - model.layers.24.self_attn.q_proj
367 - model.layers.32.self_attn.q_proj
368 - model.layers.37.self_attn.q_proj
369 - model.layers.20.self_attn.q_proj
370 - model.layers.15.self_attn.q_proj
371 - model.layers.16.self_attn.q_proj
372 - model.layers.26.self_attn.q_proj
373 - model.layers.31.self_attn.q_proj
374 # self_attn.v_proj layers
375 - model.layers.7.self_attn.v_proj
376 - model.layers.8.self_attn.v_proj
377 - model.layers.9.self_attn.v_proj
378 - model.layers.10.self_attn.v_proj
379 - model.layers.12.self_attn.v_proj
380 - model.layers.13.self_attn.v_proj
381 - model.layers.14.self_attn.v_proj
382 - model.layers.15.self_attn.v_proj
383 - model.layers.16.self_attn.v_proj
384 - model.layers.17.self_attn.v_proj
385 - model.layers.21.self_attn.v_proj
386 - model.layers.23.self_attn.v_proj
387 - model.layers.39.self_attn.v_proj
388 - model.layers.46.self_attn.v_proj
389 - model.layers.48.self_attn.v_proj
390 - model.layers.49.self_attn.v_proj
391 - model.layers.51.self_attn.v_proj
392 - model.layers.52.self_attn.v_proj
393 - model.layers.53.self_attn.v_proj
394 - model.layers.54.self_attn.v_proj
395 - model.layers.55.self_attn.v_proj
396 - model.layers.56.self_attn.v_proj
397 - model.layers.22.self_attn.v_proj
398 - model.layers.18.self_attn.v_proj
399 - model.layers.50.self_attn.v_proj
400 - model.layers.47.self_attn.v_proj
401 - model.layers.44.self_attn.v_proj
402 - model.layers.45.self_attn.v_proj
403 - model.layers.57.self_attn.v_proj
404 - model.layers.41.self_attn.v_proj
405
406| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.6265 | 0.0 | 1 | 0.6035 |
| 0.4674 | 0.25 | 327 | 0.4344 |
| 0.4337 | 0.5 | 654 | 0.4250 |
| 0.4346 | 0.75 | 981 | 0.4179 |
| 0.3985 | 1.0 | 1308 | 0.4118 |
| 0.3128 | 1.23 | 1635 | 0.4201 |
| 0.3261 | 1.48 | 1962 | 0.4157 |
| 0.3259 | 1.73 | 2289 | 0.4122 |
| 0.3126 | 1.98 | 2616 | 0.4079 |
| 0.2265 | 2.21 | 2943 | 0.4441 |
| 0.2297 | 2.46 | 3270 | 0.4427 |
| 0.2424 | 2.71 | 3597 | 0.4425 |