Views
No views yet
axolotl, with the following configuration1base_model: Qwen/Qwen2.5-VL-7B-Instruct
2processor_type: AutoProcessor
3
4# these 3 lines are needed for now to handle vision chat templates w images
5skip_prepare_dataset: true
6remove_unused_columns: false
7sample_packing: false
8
9chat_template: qwen2_vl
10datasets:
11 - path: e-zorzi/reasoning_distractors_choice_chat_only
12 type: chat_template
13 split: train
14 revision: small
15
16test_datasets:
17 - path: e-zorzi/reasoning_distractors_choice_chat_only
18 type: chat_template
19 split: val_seen[:5%]
20 revision: small
21 - path: e-zorzi/reasoning_distractors_choice_chat_only
22 type: chat_template
23 split: val_unseen[:5%]
24 revision: small
25
26load_in_8bit: True
27adapter: lora
28lora_model_dir:
29
30sequence_len: 2048 #8192
31pad_to_sequence_len: false
32
33lora_r: 128
34lora_alpha: 16
35lora_dropout: 0.05
36lora_target_modules: 'model.language_model.layers.[\d]+.(mlp|cross_attn|self_attn).(up|down|gate|q|k|v|o)_proj'
37
38gradient_accumulation_steps: 4
39micro_batch_size: 16
40num_epochs: 15
41optimizer: adamw_bnb_8bit
42lr_scheduler: cosine
43learning_rate: 0.001
44
45bf16: true
46fp16:
47tf32: true
48
49gradient_checkpointing: true
50logging_steps: 1
51flash_attention: true
52eager_attention:
53
54warmup_ratio: 0.0067
55evals_per_epoch: 3
56saves_per_epoch: 1
57save_strategy: epoch
58weight_decay: 0.0
59
60# save_first_step: true # uncomment this to validate checkpoint saving works with your config
61