Views
No views yet
1base_model: microsoft/Phi-3-mini-4k-instruct
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4trust_remote_code: True
5load_in_8bit: true
6load_in_4bit: false
7strict: false
8sequence_len: 4096
9bf16: auto
10fp16:
11tf32: false
12flash_attention: true
13# Data
14
15datasets:
16 - path: macadeliccc/opus_samantha
17 type: sharegpt
18 conversation: chatml
19
20# Iterations
21num_epochs: 3
22
23# Evaluation
24val_set_size: 0.05
25evals_per_epoch: 5
26eval_table_size:
27eval_max_new_tokens: 128
28eval_sample_packing: false
29eval_batch_size: 1
30
31# LoRA
32output_dir: ./lora-out
33adapter: lora
34lora_model_dir:
35lora_r: 32
36lora_alpha: 16
37lora_dropout: 0.05
38lora_target_linear: true
39lora_fan_in_fan_out:
40lora_target_modules:
41
42lora_modules_to_save:
43 - embed_tokens
44 - lm_head
45
46# Sampling
47sample_packing: false
48pad_to_sequence_len: false
49
50# Batching
51gradient_accumulation_steps: 4
52micro_batch_size: 4
53gradient_checkpointing: true
54
55# wandb
56wandb_project:
57
58# Optimizer
59optimizer: paged_adamw_8bit
60lr_scheduler: cosine
61learning_rate: 0.0002
62
63# Misc
64train_on_inputs: false
65group_by_length: false
66early_stopping_patience:
67resume_from_checkpoint:
68local_rank:
69logging_steps: 1
70xformers_attention:
71debug:
72deepspeed:
73weight_decay: 0
74fsdp:
75fsdp_config:
76special_tokens:
77 bos_token: "<s>"
78 eos_token: "</s>"
79 unk_token: "<unk>"
80tokens: # these are delimiters
81 - "<|im_start|>"
82 - "<|im_end|>"