Views
No views yet
0.13.0.dev01base_model: Qwen/Qwen3-8B
2# Automatically upload checkpoint and final model to HF
3# hub_model_id: username/custom_model_name
4
5quantization_config:
6 load_in_4bit: false
7 bnb_4bit_quant_type: "nf4"
8 bnb_4bit_compute_dtype: "bfloat16"
9 bnb_4bit_use_double_quant: false
10strict: false
11
12train_on_inputs: false
13
14datasets:
15 - path: /workspace/outputs/training_data/
16 ds_type: json
17 data_files:
18 - ge_v2.json
19 type: chat_template
20dataset_prepared_path:
21val_set_size: 0.00
22output_dir: /workspace/outputs/FT_v3_ge
23
24sequence_len: 1024
25sample_packing: true
26eval_sample_packing: true
27
28
29adapter: lora
30lora_model_dir:
31lora_r: 16
32lora_alpha: 32
33lora_dropout: 0.05
34lora_target_linear: true
35
36use_wandb: true
37wandb_project: "Tiboobs-GE"
38wandb_name: SFT_GE
39
40gradient_accumulation_steps: 2
41micro_batch_size: 1
42num_epochs: 3 # maybe more
43optimizer: adamw_torch_fused
44lr_scheduler: cosine
45learning_rate: 0.0002
46
47bf16: auto
48tf32: true
49
50gradient_checkpointing: true
51gradient_checkpointing_kwargs:
52 use_reentrant: false
53resume_from_checkpoint:
54logging_steps: 1
55flash_attention: true
56
57warmup_ratio: 0.1
58#evals_per_epoch: 4
59saves_per_epoch: 1
60weight_decay: 0.0
61#fsdp:
62# - full_shard
63# - auto_wrap
64#fsdp_config:
65# fsdp_limit_all_gathers: true
66# fsdp_sync_module_states: true
67# fsdp_offload_params: true
68# fsdp_use_orig_params: false
69# fsdp_cpu_ram_efficient_loading: true
70# fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
71# fsdp_transformer_layer_cls_to_wrap: Qwen3DecoderLayer
72# fsdp_state_dict_type: FULL_STATE_DICT
73# fsdp_sharding_strategy: FULL_SHARD
74special_tokens:
75
76# save_first_step: true # uncomment this to validate checkpoint saving works with your config