1base_model: mistralai/Mistral-7B-v0.1
2model_type: MistralForCausalLM
3tokenizer_type: AutoTokenizer
4
5datasets:
6 - path: /workspace/axolotl/train_model/data/finetune_dataset.jsonl
7 type:
8 field_instruction: prompt
9 field_output: response
10 format: "[INST] {instruction} [/INST]"
11 no_input_format: "[INST] {instruction} [/INST]"
12 system_prompt: "" # optionnel
13
14val_set_size: 0.05
15output_dir: /workspace/outputs-lovelace
16hub_model_id: mikefol/test
17sequence_len: 2048
18sample_packing: true
19eval_sample_packing: false
20
21micro_batch_size: 4
22gradient_accumulation_steps: 4
23num_epochs: 3
24
25optimizer: paged_adamw_32bit
26lr_scheduler: cosine
27learning_rate: 2e-5
28
29gradient_checkpointing: true
30use_wandb: false
31push_to_hub: false
32hub_private_repo: false
33trust_remote_code: true
34
35save_strategy: "no"
36save_optimizer: false
37save_safetensors: false
38