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
9hub_model_id: jeiku/completion4B
10hub_strategy: "all_checkpoints"
11push_dataset_to_hub:
12hf_use_auth_token: true
13
14datasets:
15 - path: Mielikki/Erebus-87k
16 type: completion
17 field: body
18
19shuffle_merged_datasets: true
20val_set_size: 0.0025
21output_dir: ./outputs/out
22
23adapter:
24lora_r:
25lora_alpha:
26lora_dropout:
27lora_target_linear:
28
29sequence_len: 8192
30sample_packing: true
31eval_sample_packing: false
32pad_to_sequence_len: true
33
34plugins:
35 - axolotl.integrations.liger.LigerPlugin
36liger_rope: true
37liger_rms_norm: true
38liger_swiglu: true
39liger_fused_linear_cross_entropy: true
40
41wandb_project: EXP4B
42wandb_entity:
43wandb_watch:
44wandb_name: EXP4B
45wandb_log_model:
46
47gradient_accumulation_steps: 12
48micro_batch_size: 3
49num_epochs: 1
50optimizer: adamw_bnb_8bit
51lr_scheduler: cosine
52learning_rate: 0.00001
53weight_decay: 0.05
54
55train_on_inputs: false
56group_by_length: false
57bf16: auto
58fp16:
59tf32: true
60
61gradient_checkpointing: true
62early_stopping_patience:
63resume_from_checkpoint:
64local_rank:
65logging_steps: 1
66xformers_attention:
67flash_attention: true
68
69warmup_ratio: 0.1
70evals_per_epoch: 4
71eval_table_size:
72eval_max_new_tokens: 128
73saves_per_epoch: 1
74
75debug:
76deepspeed: deepspeed_configs/zero3_bf16.json
77fsdp:
78fsdp_config:
79
80special_tokens:
81 pad_token: <|finetune_right_pad_id|>
1base_model: jeiku/completion4B
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8
9hub_model_id: jeiku/instructered4B
10hub_strategy: "all_checkpoints"
11push_dataset_to_hub:
12hf_use_auth_token: true
13
14datasets:
15 - path: FourOhFour/Instruct_Phase
16 type: sharegpt
17 conversation: chatml
18
19chat_template: chatml
20
21shuffle_merged_datasets: true
22val_set_size: 0.0025
23output_dir: ./outputs/out
24
25adapter:
26lora_r:
27lora_alpha:
28lora_dropout:
29lora_target_linear:
30
31sequence_len: 8192
32sample_packing: true
33eval_sample_packing: false
34pad_to_sequence_len: true
35
36plugins:
37 - axolotl.integrations.liger.LigerPlugin
38liger_rope: true
39liger_rms_norm: true
40liger_swiglu: true
41liger_fused_linear_cross_entropy: true
42
43wandb_project: EXP4B
44wandb_entity:
45wandb_watch:
46wandb_name: EXP4B
47wandb_log_model:
48
49gradient_accumulation_steps: 12
50micro_batch_size: 3
51num_epochs: 2
52optimizer: adamw_bnb_8bit
53lr_scheduler: cosine
54learning_rate: 0.00001
55weight_decay: 0.05
56
57train_on_inputs: false
58group_by_length: false
59bf16: auto
60fp16:
61tf32: true
62
63gradient_checkpointing: true
64early_stopping_patience:
65resume_from_checkpoint:
66local_rank:
67logging_steps: 1
68xformers_attention:
69flash_attention: true
70
71warmup_ratio: 0.1
72evals_per_epoch: 4
73eval_table_size:
74eval_max_new_tokens: 128
75saves_per_epoch: 2
76
77debug:
78deepspeed: deepspeed_configs/zero3_bf16.json
79fsdp:
80fsdp_config:
81
82special_tokens:
83 pad_token: <|finetune_right_pad_id|>
1base_model: jeiku/instructered4B
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8
9hub_model_id: jeiku/TheBest4B
10hub_strategy: "all_checkpoints"
11push_dataset_to_hub:
12hf_use_auth_token: true
13
14datasets:
15 - path: FourOhFour/RP_Phase
16 type: sharegpt
17 conversation: chatml
18
19chat_template: chatml
20
21shuffle_merged_datasets: true
22val_set_size: 0.0025
23output_dir: ./outputs/out
24
25adapter:
26lora_r:
27lora_alpha:
28lora_dropout:
29lora_target_linear:
30
31sequence_len: 8192
32sample_packing: true
33eval_sample_packing: false
34pad_to_sequence_len: true
35
36plugins:
37 - axolotl.integrations.liger.LigerPlugin
38liger_rope: true
39liger_rms_norm: true
40liger_swiglu: true
41liger_fused_linear_cross_entropy: true
42
43wandb_project: EXP4B
44wandb_entity:
45wandb_watch:
46wandb_name: EXP4B
47wandb_log_model:
48
49gradient_accumulation_steps: 12
50micro_batch_size: 3
51num_epochs: 2
52optimizer: adamw_bnb_8bit
53lr_scheduler: cosine
54learning_rate: 0.00001
55weight_decay: 0.05
56
57train_on_inputs: false
58group_by_length: false
59bf16: auto
60fp16:
61tf32: true
62
63gradient_checkpointing: true
64early_stopping_patience:
65resume_from_checkpoint:
66local_rank:
67logging_steps: 1
68xformers_attention:
69flash_attention: true
70
71warmup_ratio: 0.1
72evals_per_epoch: 4
73eval_table_size:
74eval_max_new_tokens: 128
75saves_per_epoch: 2
76
77debug:
78deepspeed: deepspeed_configs/zero3_bf16.json
79fsdp:
80fsdp_config:
81
82special_tokens:
83 pad_token: <|finetune_right_pad_id|>
1base_model: FourOhFour/Crispy_Crab_4B
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8
9hub_model_id: jeiku/aura4bkto
10hub_strategy: "all_checkpoints"
11push_dataset_to_hub:
12hf_use_auth_token: true
13
14chat_template: chatml
15
16rl: kto
17rl_beta: 0.2
18kto_desirable_weight: 0.2
19
20datasets:
21 - path: anthracite-core/full-opus-chosen-hermes-rejected-kto-v1
22 type: chatml.argilla
23
24shuffle_merged_datasets: true
25val_set_size: 0.0
26output_dir: ./outputs/out
27
28sequence_len: 8192
29sample_packing: false
30eval_sample_packing: false
31pad_to_sequence_len: false
32
33wandb_project: Aura-4B
34wandb_entity:
35wandb_watch:
36wandb_name: Aura-4B
37wandb_log_model:
38
39gradient_accumulation_steps: 16
40micro_batch_size: 2
41num_epochs: 2
42max_steps: 500
43
44optimizer: adamw_8bit
45lr_scheduler: cosine
46learning_rate: 0.00001
47weight_decay: 0.05
48
49train_on_inputs: false
50group_by_length: false
51bf16: auto
52fp16:
53tf32: true
54
55gradient_checkpointing: true
56gradient_checkpointing_kwargs:
57 use_reentrant: true
58remove_unused_columns: false
59early_stopping_patience:
60resume_from_checkpoint:
61local_rank:
62logging_steps: 1
63xformers_attention:
64flash_attention: true
65
66warmup_steps: 10
67evals_per_epoch: 2
68eval_table_size:
69eval_max_new_tokens:
70saves_per_epoch: 1
71
72debug:
73deepspeed:
74fsdp:
75fsdp_config:
76fsdp:
77fsdp_config:
78
79special_tokens:
80 pad_token: <|finetune_right_pad_id|>