1base_model: "Workermen/ru_qwen"
2model_type: Qwen2ForCausalLM
3tokenizer_type: AutoTokenizer
4
5plugins:
6 - axolotl.integrations.liger.LigerPlugin
7liger_rope: true
8liger_rms_norm: true
9liger_glu_activation: true
10liger_fused_linear_cross_entropy: true
11
12load_in_8bit: false
13load_in_4bit: false
14strict: false
15
16chat_template: chatml
17
18datasets:
19 - path: "/workspace/data/axolotl/examples/qwen2/train"
20 data_files: "./train/train_01.jsonl"
21 type: chat_template
22 field_messages: conversations
23
24 - path: "/workspace/data/axolotl/examples/qwen2/train"
25 data_files: "./train/train_02.jsonl"
26 type: chat_template
27 field_messages: conversations
28
29 - path: "/workspace/data/axolotl/examples/qwen2/train"
30 data_files: "./train/train_03.jsonl"
31 type: chat_template
32 field_messages: conversations
33
34 - path: "/workspace/data/axolotl/examples/qwen2/train"
35 data_files: "./train/train_04.jsonl"
36 type: chat_template
37 field_messages: conversations
38
39 - path: "/workspace/data/axolotl/examples/qwen2/train"
40 data_files: "./train/validation.jsonl"
41 type: chat_template
42 field_messages: conversations
43
44
45dataset_prepared_path: "/workspace/data/axolotl/examples/qwen2"
46val_set_size: 0.05
47output_dir: "/workspace/data/pyx_03_15_1_tochno"
48
49sequence_len: 5300
50sample_packing: false
51pad_to_sequence_len: true
52
53#mlflow_tracking_uri: "http://localhost:8080"
54#mlflow_experiment_name: "my_axolotl_experiment"
55#mlflow_run_name: "run-1"
56#hf_mlflow_log_artifacts: true
57
58wandb_project:
59wandb_entity:
60wandb_watch:
61wandb_name:
62wandb_log_model:
63
64gradient_accumulation_steps: 32
65micro_batch_size: 4
66num_epochs: 2
67
68optimizer: paged_adamw_8bit
69lr_scheduler: cosine
70cosine_min_lr_ratio: 0.5
71cosine_constant_lr_ratio: 0.5
72learning_rate: 0.00004
73
74train_on_inputs: false
75group_by_length: false
76bf16: auto
77fp16:
78tf32: false
79max_grad_norm: 2
80
81gradient_checkpointing: true
82gradient_checkpointing_kwargs:
83 use_reentrant: false
84early_stopping_patience:
85resume_from_checkpoint:
86logging_steps: 1
87xformers_attention:
88flash_attention: true
89
90save_safetensors: true
91
92warmup_steps: 10
93evals_per_epoch: 4
94eval_table_size:
95saves_per_epoch: 1
96debug:
97deepspeed: #deepspeed_configs/zero3_bf16_cpuoffload_all.json
98weight_decay: 0.0
99fsdp:
100# - full_shard
101# - auto_wrap
102fsdp_config:
103# fsdp_limit_all_gathers: true
104# fsdp_sync_module_states: true
105# fsdp_offload_params: true
106# fsdp_use_orig_params: false
107# fsdp_cpu_ram_efficient_loading: true
108# fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
109# fsdp_transformer_layer_cls_to_wrap: LlamaDecoderLayer
110# fsdp_state_dict_type: FULL_STATE_DICT
111# fsdp_sharding_strategy: FULL_SHARD
112# fsdp_backward_prefetch: BACKWARD_PRE
113special_tokens:
114