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