1base_model: mayflowergmbh/occiglot-10b-de-en-instruct
2model_type: MistralForCausalLM
3tokenizer_type: LlamaTokenizer
4is_mistral_derived_model: true
5
6load_in_8bit: false
7load_in_4bit: true
8strict: false
9
10rl: dpo
11datasets:
12 - path: johannhartmann/mistralorpo
13 split: train
14 type: chatml.intel
15dataset_prepared_path: last_run_prepared
16val_set_size: 0.0
17output_dir: ./occiglot10b/
18save_total_limit: 3
19
20adapter: qlora
21lora_model_dir:
22
23sequence_len: 2048
24sample_packing: false
25pad_to_sequence_len: false
26
27lora_r: 64
28lora_alpha: 32
29lora_dropout: 0.05
30lora_target_linear: true
31lora_modules_to_save:
32 - embed_tokens
33 - lm_head
34lora_fan_in_fan_out:
35lora_target_modules:
36 - gate_proj
37 - down_proj
38 - up_proj
39 - q_proj
40 - v_proj
41 - k_proj
42 - o_proj
43
44wandb_project: occiglot
45wandb_entity: mayflowerteam
46wandb_watch:
47wandb_run_id:
48wandb_log_model:
49
50gradient_accumulation_steps: 4
51micro_batch_size: 2
52num_epochs: 1
53optimizer: paged_adamw_8bit
54adam_beta2: 0.95
55adam_epsilion: 0.00001
56lr_scheduler: cosine
57learning_rate: 2e-5
58
59train_on_inputs: false
60group_by_length: false
61bf16: true
62fp16: false
63tf32: false
64
65gradient_checkpointing: true
66early_stopping_patience:
67resume_from_checkpoint:
68local_rank:
69logging_steps: 1
70xformers_attention:
71flash_attention: true
72
73warmup_steps: 100
74eval_steps:
75eval_table_size:
76eval_table_max_new_tokens: 128
77save_steps: 239
78debug:
79deepspeed:
80weight_decay: 0.1
81fsdp:
82fsdp_config:
83special_tokens:
84save_safetensors: true
85special_tokens:
86 eos_token: "<|im_end|>"
87tokens: # these are delimiters
88 - "<|im_start|>"
89chat_template: chatml
90