1base_model: Qwen/Qwen2.5-7B-Instruct
2trust_remote_code: true
3model_type: AutoModelForCausalLM
4tokenizer_type: AutoTokenizer
5
6load_in_8bit:
7load_in_4bit:
8strict: false
9
10datasets:
11 - path: medalpaca/medical_meadow_medqa
12 type: alpaca
13dataset_prepared_path:
14val_set_size: 0.1
15output_dir: ./sft-qwen25
16
17sequence_len: 8192
18sample_packing: true
19eval_sample_packing: true
20pad_to_sequence_len: true
21
22
23wandb_project: sft-qwen-25-7b-instruct
24wandb_entity:
25wandb_watch:
26wandb_name:
27wandb_log_model:
28
29gradient_accumulation_steps: 1
30micro_batch_size: 1
31num_epochs: 1
32optimizer: adamw_torch
33lr_scheduler: cosine
34learning_rate: 0.00001
35
36train_on_inputs: false
37group_by_length: false
38bf16: true
39fp16: false
40tf32: false
41
42gradient_checkpointing: true
43
44logging_steps: 1
45xformers_attention:
46flash_attention: true
47
48warmup_steps:
49eval_steps: 10
50save_steps: 40
51
52evals_per_epoch:
53saves_per_epoch:
54
55debug:
56deepspeed: deepspeed_configs/zero2.json
57weight_decay:
58fsdp:
59fsdp_config:
60special_tokens:
61
62hub_model_id: neginashz/sft-qwen-25-7b-instruct
63hub_strategy: all_checkpoints
64early_stopping_patience: 3
65
66auto_resume_from_checkpoints: true
67
68
69