base_model: Qwen/Qwen2.5-1.5B-Instruct
model_type: Qwen2ForCausalLM
tokenizer_type: AutoTokenizer

load_in_8bit: true
load_in_4bit: false

datasets:
  - path: cfierro/gcd  
    split: half_train_instr_q  # 3.81k
    type: chat_template
  - path: cfierro/gcd
    split: half_train_sycophancy_correct  # 3.81k
    type: chat_template
  - path: cfierro/pv-prompts-non-sycophantic_Qwen2.5-1.5B-Instruct
    type: chat_template
    split: train_6x
dataset_prepared_path: /workspace/axolotl-datasets/Qwen2.5-1.5B-Instruct/gcd_sycophancy_joint_pv_prompts_6x
val_set_size: 0.05
output_dir: /workspace/axolotl-outputs/Qwen2.5-1.5B-Instruct-gcd_sycophancy_joint_pv_prompts_6x

sequence_len: 4096
sample_packing: false
eval_sample_packing: false
pad_to_sequence_len: true

adapter: lora
lora_model_dir:
lora_r: 32
lora_alpha: 64
lora_dropout: 0.00
lora_target_linear: true
lora_modules_to_save:
  - embed_tokens
  - lm_head
merge_lora: true

wandb_project: weight-diff-ft
wandb_entity: cfierro
wandb_watch: all
wandb_name: Qwen2.5-1.5B-Instruct-gcd_sycophancy_joint_pv_prompts_6x
wandb_log_model: "false"
#hub_model_id: coastalcph/Qwen2.5-1.5B-Instruct-gcd_sycophancy_joint_pv_prompts_6x

gradient_accumulation_steps: 8
micro_batch_size: 2
num_train_epochs: 1
early_stopping_patience: 2
optimizer: adamw_8bit
lr_scheduler: linear
learning_rate: 2e-4
weight_decay: 0.01
warmup_steps: 5

bf16: auto
tf32: false

gradient_checkpointing: true
resume_from_checkpoint:
logging_steps: 1
flash_attention: true

eval_steps: 60 
save_steps: 60
save_total_limit: 1
load_best_model_at_end: true
special_tokens: