1adapter: lora
2base_model: unsloth/Meta-Llama-3.1-8B-Instruct
3bf16: true
4fp16: false
5
6load_in_4bit: true
7load_in_8bit: false
8
9datasets:
10 - path: mx003/cve_dataset_1000
11 type: chat_template
12 field_messages: messages
13 chat_template: llama3
14
15lora_r: 32
16lora_alpha: 64
17lora_dropout: 0.05
18lora_target_modules:
19 - q_proj
20 - v_proj
21 - k_proj
22 - o_proj
23 - gate_proj
24 - down_proj
25 - up_proj
26
27lora_modules_to_save:
28 - embed_tokens
29 - lm_head
30
31gradient_accumulation_steps: 4
32gradient_checkpointing: true
33micro_batch_size: 4
34num_epochs: 2
35learning_rate: 0.0002
36optimizer: adamw_torch_fused
37
38output_dir: ./outputs/mymodel
39sequence_len: 4096
40save_steps: 50
41flash_attention: true
42sample_packing: true
43group_by_length: true
44train_on_inputs: false
45
46special_tokens:
47 pad_token: <|end_of_text|>
48tokens:
49 - "<|end_of_text|>"