1base_model: Delta-Vector/Holland-4B
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8
9datasets:
10 - path: NewEden/xlam-function-calling-60k-shareGPT
11 type: sharegpt
12 conversation: chatml
13 - path: gardner/glaive-function-calling-v2-sharegpt
14 type: sharegpt
15 conversation: chatml
16
17chat_template: chatml
18
19val_set_size: 0.01
20output_dir: ./outputs/out
21
22adapter:
23lora_r:
24lora_alpha:
25lora_dropout:
26lora_target_linear:
27
28sequence_len: 8192
29# sequence_len: 32768
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: GnX Func Calling
42wandb_entity:
43wandb_watch:
44wandb_name: Func Calling GnX
45wandb_log_model:
46
47gradient_accumulation_steps: 32
48micro_batch_size: 1
49num_epochs: 2
50optimizer: adamw_bnb_8bit
51lr_scheduler: cosine
52learning_rate: 0.00002
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: /workspace/axolotl/deepspeed_configs/zero2.json
77fsdp:
78fsdp_config:
79
80special_tokens:
81 pad_token: <|finetune_right_pad_id|>
82