Views
No views yet
0.12.0.dev01base_model: Qwen/Qwen2.5-7B-Instruct
2# optionally might have model_type or tokenizer_type
3model_type: Qwen2ForCausalLM
4tokenizer_type: AutoTokenizer
5# Automatically upload checkpoint and final model to HF
6# hub_model_id: username/custom_model_name
7
8load_in_8bit: true
9load_in_4bit: false
10
11datasets:
12 - path: cfierro/alpaca-en2fr
13 type: alpaca
14dataset_prepared_path: /workspace/axolotl-datasets/Qwen2.5-7B/en2fr_alpaca
15val_set_size: 0.02
16output_dir: /workspace/axolotl-outputs/Qwen2.5-7B-en2fr_alpaca-lora
17
18sequence_len: 4096
19sample_packing: true
20eval_sample_packing: false
21pad_to_sequence_len: true
22
23adapter: lora
24lora_model_dir:
25lora_r: 32
26lora_alpha: 16
27lora_dropout: 0.05
28lora_target_linear: true
29lora_modules_to_save:
30 - embed_tokens
31 - lm_head
32merge_lora: true
33
34wandb_project: weight-diff-ft
35wandb_entity: cfierro
36wandb_watch: all
37wandb_name: Qwen2.5-7B-en2fr_alpaca-lora
38wandb_log_model: "false"
39hub_model_id: coastalcph/Qwen2.5-7B-en2fr_alpaca-lora
40
41gradient_accumulation_steps: 4
42micro_batch_size: 2
43max_steps: 1000
44optimizer: adamw_bnb_8bit
45lr_scheduler: cosine
46learning_rate: 0.0002
47
48bf16: auto
49tf32: false
50
51gradient_checkpointing: true
52resume_from_checkpoint:
53logging_steps: 1
54flash_attention: true
55
56warmup_steps: 10
57early_stopping_patience: 2
58eval_steps: 60
59save_steps: 60
60save_total_limit: 1
61load_best_model_at_end: true
62weight_decay: 0.0
63special_tokens:
64| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0 | 0 | 2.3403 |
| 1.1371 | 0.32 | 60 | 1.0314 |
| 1.0978 | 0.64 | 120 | 1.0035 |
| 1.004 | 0.96 | 180 | 0.9847 |
| 0.8761 | 1.2773 | 240 | 0.9951 |
| 0.8661 | 1.5973 | 300 | 0.9886 |