MaziyarPanahi commited on
Commit
e179d94
1 Parent(s): a690ebe

Create axolotl-dpo.yaml (#6)

Browse files

- Create axolotl-dpo.yaml (28e29a8b1e945b5cb8c27d3748d86df1673ef114)

Files changed (1) hide show
  1. axolotl-dpo.yaml +75 -0
axolotl-dpo.yaml ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: meta-llama/Meta-Llama-3-8B-Instruct
2
+ model_type: AutoModelForCausalLM
3
+ tokenizer_type: AutoTokenizer
4
+
5
+ load_in_8bit: false
6
+ load_in_4bit: true
7
+ strict: false
8
+
9
+ save_safetensors: true
10
+
11
+ rl: dpo
12
+ chat_template: chatml
13
+ datasets:
14
+ - path: Intel/orca_dpo_pairs
15
+ split: train
16
+ type: chatml.intel
17
+
18
+ dataset_prepared_path:
19
+ val_set_size: 0.1
20
+ output_dir: ./models/Meta-Llama-3-8B-instruct-DPO-v0.3
21
+
22
+ adapter: qlora
23
+ lora_model_dir:
24
+
25
+ sequence_len: 8192
26
+ sample_packing: false
27
+ pad_to_sequence_len: false
28
+
29
+ lora_r: 64
30
+ lora_alpha: 32
31
+ lora_dropout: 0.05
32
+ lora_target_linear: true
33
+ lora_fan_in_fan_out:
34
+ lora_modules_to_save:
35
+ - embed_tokens
36
+ - lm_head
37
+
38
+ wandb_project: llama-3-instruct-dpo
39
+ wandb_entity:
40
+ wandb_watch:
41
+ wandb_name:
42
+ wandb_log_model:
43
+
44
+ gradient_accumulation_steps: 2
45
+ micro_batch_size: 1
46
+ num_epochs: 3
47
+ optimizer: paged_adamw_32bit
48
+ lr_scheduler: cosine
49
+ learning_rate: 5e-6
50
+ train_on_inputs: false
51
+ group_by_length: false
52
+
53
+ bf16: auto
54
+ fp16:
55
+ tf32:
56
+
57
+ gradient_checkpointing: true
58
+ early_stopping_patience:
59
+ resume_from_checkpoint:
60
+ local_rank:
61
+ logging_steps: 1
62
+ xformers_attention:
63
+ flash_attention: true
64
+ warmup_steps: 100
65
+ evals_per_epoch: 1
66
+ eval_table_size:
67
+ eval_table_max_new_tokens: 128
68
+ saves_per_epoch: 6
69
+ debug:
70
+ deepspeed:
71
+ weight_decay: 0.0
72
+ fsdp:
73
+ fsdp_config:
74
+ special_tokens:
75
+ pad_token: "<|end_of_text|>"