1base_model: IntervitensInc/Llama-3.1-Minitron-4B-Width-Base-chatml
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8
9datasets:
10 - path: NewEden/Gryphe-3.5-16k-Subset
11 type: sharegpt
12 conversation: chatml
13 - path: NewEden/Claude-Instruct-5k
14 type: sharegpt
15 conversation: chatml
16 - path: NewEden/kaloisazasedhandsomefurry
17 type: sharegpt
18 conversation: chatml
19 - path: lodrick-the-lafted/NopmWritingStruct
20 type: sharegpt
21 conversation: chatml
22 - path: PocketDoc/Dans-MemoryCore-CoreCurriculum-Small
23 type: sharegpt
24 conversation: chatml
25 - path: Epiculous/Synthstruct-Gens-v1.1-Filtered-n-Cleaned
26 type: sharegpt
27 conversation: chatml
28 - path: anthracite-org/kalo-opus-instruct-22k-no-refusal
29 type: sharegpt
30 conversation: chatml
31 - path: PJMixers/lodrick-the-lafted_OpusStories-ShareGPT
32 type: sharegpt
33 conversation: chatml
34 - path: NewEden/Stheno-Data-filtered-8k-subset
35 type: sharegpt
36 conversation: chatml
37 - path: Epiculous/SynthRP-Gens-v1.1-Filtered-n-Cleaned
38 type: sharegpt
39 conversation: chatml
40 - path: lodrick-the-lafted/kalo-opus-instruct-3k-filtered
41 type: sharegpt
42 conversation: chatml
43chat_template: chatml
44
45val_set_size: 0.01
46output_dir: ./outputs/out
47
48adapter:
49lora_r:
50lora_alpha:
51lora_dropout:
52lora_target_linear:
53
54sequence_len: 16384
55# sequence_len: 32768
56sample_packing: true
57eval_sample_packing: false
58pad_to_sequence_len: true
59
60plugins:
61 - axolotl.integrations.liger.LigerPlugin
62liger_rope: true
63liger_rms_norm: true
64liger_swiglu: true
65liger_fused_linear_cross_entropy: true
66
67wandb_project: HollandV2
68wandb_entity:
69wandb_watch:
70wandb_name: hollandV2
71wandb_log_model:
72
73gradient_accumulation_steps: 32
74micro_batch_size: 1
75num_epochs: 2
76optimizer: adamw_bnb_8bit
77lr_scheduler: cosine
78learning_rate: 0.00001
79weight_decay: 0.05
80
81train_on_inputs: false
82group_by_length: false
83bf16: auto
84fp16:
85tf32: true
86
87gradient_checkpointing: true
88early_stopping_patience:
89resume_from_checkpoint:
90local_rank:
91logging_steps: 1
92xformers_attention:
93flash_attention: true
94
95warmup_ratio: 0.1
96evals_per_epoch: 4
97eval_table_size:
98eval_max_new_tokens: 128
99saves_per_epoch: 1
100
101debug:
102deepspeed: /workspace/axolotl/deepspeed_configs/zero2.json
103fsdp:
104fsdp_config:
105
106special_tokens:
107 pad_token: <|finetune_right_pad_id|>
108
109