1base_model: microsoft/phi-2
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8
9datasets:
10 - path: Intel/orca_dpo_pairs
11 type:
12 system_prompt: ""
13 field_system: system
14 field_instruction: question
15 field_output: rejected
16 field_output: chosen
17
18dataset_prepared_path:
19val_set_size: 0.05
20output_dir: ./phi-sft-out
21
22sequence_len: 2048
23sample_packing: true
24pad_to_sequence_len: true
25
26adapter:
27lora_model_dir:
28lora_r:
29lora_alpha:
30lora_dropout:
31lora_target_linear:
32lora_fan_in_fan_out:
33
34wandb_project:
35wandb_entity:
36wandb_watch:
37wandb_name:
38wandb_log_model:
39
40gradient_accumulation_steps: 1
41micro_batch_size: 2
42num_epochs: 2
43optimizer: adamw_torch
44adam_beta2: 0.95
45adam_epsilon: 0.00001
46max_grad_norm: 1.0
47lr_scheduler: cosine
48learning_rate: 0.000003
49
50train_on_inputs: false
51group_by_length: false
52bf16: auto
53fp16:
54tf32: true
55
56gradient_checkpointing: true
57gradient_checkpointing_kwargs:
58 use_reentrant: True
59early_stopping_patience:
60resume_from_checkpoint:
61local_rank:
62logging_steps: 1
63xformers_attention:
64flash_attention: true
65
66warmup_steps: 100
67evals_per_epoch: 4
68saves_per_epoch: 1
69debug:
70deepspeed:
71weight_decay: 0.1
72fsdp:
73fsdp_config:
74resize_token_embeddings_to_32x: true
75special_tokens:
76 pad_token: "<|endoftext|>"
77