1base_model: cognitivecomputations/dolphin-2.6-mistral-7b-dpo-laser
2model_type: MistralForCausalLM
3tokenizer_type: LlamaTokenizer
4
5load_in_8bit: true
6load_in_4bit: false
7strict: false
8
9datasets:
10 - path: ./julia/data.jsonl
11 type: sharegpt
12 conversation: chatml
13dataset_prepared_path: ./julia/prepared_data
14chat_template: chatml
15val_set_size: 0.05
16output_dir: ./julia/lora-out
17hub_model_id: animmina/cheater-7b
18hub_strategy: every_save
19hf_use_auth_token: true
20
21sequence_len: 2048
22sample_packing: true
23pad_to_sequence_len: true
24eval_sample_packing: false
25
26adapter: lora
27lora_model_dir:
28lora_r: 8
29lora_alpha: 16
30lora_dropout: 0.05
31lora_target_linear: true
32lora_target_modules:
33 - gate_proj
34 - down_proj
35 - up_proj
36 - q_proj
37 - v_proj
38 - k_proj
39 - o_proj
40
41wandb_project: cheater-7b
42wandb_entity:
43wandb_watch:
44wandb_name: v02
45wandb_log_model:
46
47gradient_accumulation_steps: 1
48micro_batch_size: 4
49num_epochs: 4
50optimizer: adamw_bnb_8bit
51lr_scheduler: cosine
52learning_rate: 0.00003
53
54train_on_inputs: false
55group_by_length: false
56bf16: auto
57fp16:
58tf32: false
59
60gradient_checkpointing: true
61early_stopping_patience:
62resume_from_checkpoint:
63local_rank:
64logging_steps: 1
65xformers_attention:
66flash_attention: true
67
68warmup_steps: 10
69evals_per_epoch: 4
70eval_table_size:
71eval_max_new_tokens: 128
72saves_per_epoch: 1
73debug:
74deepspeed:
75weight_decay: 0.0
76fsdp:
77fsdp_config:
78special_tokens:
79 bos_token: "<s>"
80 eos_token: "<|im_end|>"
81 unk_token: "<unk>"
82