Views
No views yet
0.8.0.dev01base_model: Qwen/Qwen2.5-7B-Instruct
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4trust_remote_code: true
5
6load_in_8bit: false
7load_in_4bit: true
8strict: false
9
10datasets:
11 - path: ugaoo/initial_advice_dataset_doctor_patient
12 type: alpaca
13val_set_size: 0
14output_dir: ./out/Qwen_Qwen2.5_7B_Instruct_ugaoo_initial_advice_dataset_doctor_patient
15
16sequence_len: 4000
17sample_packing: true
18pad_to_sequence_len: true
19
20adapter: qlora
21lora_r: 256
22lora_alpha: 512
23lora_dropout: 0.05
24lora_target_linear: true
25lora_target_modules:
26 - q_proj
27 - k_proj
28 - v_proj
29 - o_proj
30 - up_proj
31 - down_proj
32 - gate_proj
33
34wandb_project: cosmosearch
35wandb_entity:
36wandb_watch:
37wandb_name: Qwen_Qwen2.5_7B_Instruct_ugaoo_initial_advice_dataset_doctor_patient
38wandb_log_model:
39
40gradient_accumulation_steps: 3
41micro_batch_size: 4
42num_epochs: 6
43optimizer: adamw_torch
44lr_scheduler: cosine
45learning_rate: 5e-6
46
47train_on_inputs: false
48group_by_length: false
49bf16: auto
50fp16: false
51tf32: false
52
53gradient_checkpointing: true
54early_stopping_patience:
55resume_from_checkpoint:
56logging_steps: 1
57xformers_attention:
58flash_attention: true
59
60warmup_steps: 100
61evals_per_epoch: 6
62eval_table_size:
63saves_per_epoch: 1
64debug:
65deepspeed:
66weight_decay: 0.0
67fsdp:
68fsdp_config:
69save_total_limit: 6