1base_model: ConvexAI/Metabird-7B
2model_type: MistralForCausalLM
3tokenizer_type: LlamaTokenizer
4is_mistral_derived_model: true
5
6load_in_8bit: false
7load_in_4bit: false
8strict: false
9
10datasets:
11 - path: Intel/orca_dpo_pairs
12 type:
13 system_prompt: ""
14 field_system: system
15 field_instruction: question
16 field_output: rejected
17 field_output: chosen
18 format: "[INST] {instruction} [/INST]"
19 no_input_format: "[INST] {instruction} [/INST]"
20dataset_prepared_path:
21val_set_size: 0.05
22output_dir: ./out
23
24sequence_len: 8192
25sample_packing: true
26pad_to_sequence_len: true
27eval_sample_packing: false
28
29wandb_project:
30wandb_entity:
31wandb_watch:
32wandb_name:
33wandb_log_model:
34
35gradient_accumulation_steps: 4
36micro_batch_size: 2
37num_epochs: 2
38optimizer: adamw_bnb_8bit
39lr_scheduler: cosine
40learning_rate: 0.000005
41
42train_on_inputs: false
43group_by_length: false
44bf16: auto
45fp16:
46tf32: false
47
48gradient_checkpointing: true
49early_stopping_patience:
50resume_from_checkpoint:
51local_rank:
52logging_steps: 1
53xformers_attention:
54flash_attention: true
55
56warmup_steps: 10
57evals_per_epoch: 4
58eval_table_size:
59eval_table_max_new_tokens: 128
60saves_per_epoch: 1
61debug:
62deepspeed:
63weight_decay: 0.0
64fsdp:
65fsdp_config:
66special_tokens:
67 bos_token: "<s>"
68 eos_token: "</s>"
69 unk_token: "<unk>"