Views
No views yet
0.13.0.dev01adapter: lora
2base_model: meta-llama/Llama-3.1-8B-Instruct
3bf16: true
4fp16: false
5
6datasets:
7 - path: mx003/cve
8 type: chat_template
9 field_messages: messages
10 chat_template: llama3
11
12lora_r: 32
13lora_alpha: 64
14lora_dropout: 0.05
15lora_target_modules:
16 - q_proj
17 - v_proj
18 - k_proj
19 - o_proj
20 - gate_proj
21 - down_proj
22 - up_proj
23
24gradient_accumulation_steps: 4
25gradient_checkpointing: true
26micro_batch_size: 2
27num_epochs: 3
28learning_rate: 0.0002
29optimizer: adamw_torch_fused
30train_on_inputs: false
31group_by_length: true
32
33output_dir: ./outputs/mymodel
34sequence_len: 4096
35save_steps: 50
36flash_attention: true
37sample_packing: true
38special_tokens:
39 pad_token: <|end_of_text|>
40tokens:
41 - "<|end_of_text|>"
42