1adapter: lora
2base_model: unsloth/Meta-Llama-3.1-8B-Instruct
3bf16: true
4fp16: false
5
6datasets:
7 - path: mx003/cve
8 type: chat_template
9 field_messages: messages
10
11lora_r: 32
12lora_alpha: 64
13lora_dropout: 0.05
14lora_target_modules:
15 - q_proj
16 - v_proj
17 - k_proj
18 - o_proj
19 - gate_proj
20 - down_proj
21 - up_proj
22
23gradient_accumulation_steps: 4
24gradient_checkpointing: true
25micro_batch_size: 2
26num_epochs: 3
27learning_rate: 0.0002
28optimizer: adamw_torch
29train_on_inputs: false
30group_by_length: true
31
32output_dir: ./outputs/mymodel
33sequence_len: 4096
34save_steps: 50
35flash_attention: true
36sample_packing: true