Views
No views yet
0.6.01base_model: deepseek-ai/DeepSeek-R1-Distill-Qwen-7B
2
3load_in_8bit: false
4load_in_4bit: false
5
6plugins:
7 - axolotl.integrations.liger.LigerPlugin
8liger_rope: true
9liger_rms_norm: true
10liger_glu_activation: true
11liger_fused_linear_cross_entropy: true
12
13#unsloth_lora_mlp: true
14#unsloth_lora_qkv: true
15#unsloth_lora_o: true
16
17strict: false
18
19adapter: lora
20lora_r: 16
21lora_alpha: 32
22lora_dropout: 0.25
23lora_target_modules:
24 - q_proj
25 - v_proj
26 - k_proj
27 - o_proj
28lora_target_linear: true
29peft_layers_to_transform:
30loraplus_lr_ratio: 16
31
32chat_template: tokenizer_default
33datasets:
34 - path: Fizzarolli/inkmix-v2
35 type: chat_template
36 chat_template: tokenizer_default
37 split: train
38 field_messages: conversations
39 message_field_role: from
40 message_field_content: value
41
42dataset_prepared_path: last_run_prepared
43#val_set_size: 0.02
44output_dir: ./ckpts
45
46sequence_len: 8192
47sample_packing: true
48pad_to_sequence_len: true
49
50#wandb_project: teleut-7b-rp
51#wandb_entity:
52#wandb_watch:
53#wandb_name:
54#wandb_log_model: checkpoint
55
56# mlflow configuration if you're using it
57mlflow_tracking_uri: https://public-tracking.mlflow-e00zzfjq11ky6jcgtv.backbone-e00bgn6e63256prmhq.msp.eu-north1.nebius.cloud
58mlflow_experiment_name: deepseek-r1-7b-rp-inkmixv2
59mlflow_run_name: v1
60hf_mlflow_log_artifacts: false
61
62gradient_accumulation_steps: 1
63micro_batch_size: 16
64num_epochs: 2
65optimizer: paged_adamw_8bit
66lr_scheduler: cosine
67learning_rate: 6e-5
68
69train_on_inputs: false
70group_by_length: false
71bf16: auto
72fp16:
73tf32: false
74
75gradient_checkpointing: unsloth
76gradient_checkpointing_kwargs:
77 use_reentrant: false
78early_stopping_patience:
79resume_from_checkpoint:
80logging_steps: 1
81xformers_attention:
82flash_attention: true
83
84#deepspeed: deepspeed_configs/zero3_bf16.json
85
86warmup_steps: 25
87#evals_per_epoch: 4
88eval_table_size:
89saves_per_epoch: 10
90debug:
91weight_decay: 0.01
92