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 - knowledge_v2.json
19 - drug_v2.json
20 - ge_v2.json
21 type: chat_template
22dataset_prepared_path:
23val_set_size: 0.00
24output_dir: /workspace/outputs/FT_fused
25
26sequence_len: 512
27sample_packing: true
28eval_sample_packing: true
29
30
31adapter: lora
32lora_model_dir:
33lora_r: 16
34lora_alpha: 32
35lora_dropout: 0.05
36lora_target_linear: true
37
38use_wandb: true
39wandb_project: "Tiboobs_Fused"
40wandb_name: SFT_Test
41
42gradient_accumulation_steps: 2
43micro_batch_size: 1
44num_epochs: 3 # maybe more
45optimizer: adamw_torch_fused
46lr_scheduler: cosine
47learning_rate: 0.0002
48
49bf16: auto
50tf32: true
51
52gradient_checkpointing: true
53gradient_checkpointing_kwargs:
54 use_reentrant: false
55resume_from_checkpoint:
56logging_steps: 1
57flash_attention: true
58
59warmup_ratio: 0.1
60#evals_per_epoch: 4
61saves_per_epoch: 1
62weight_decay: 0.0
63#fsdp:
64# - full_shard
65# - auto_wrap
66#fsdp_config:
67# fsdp_limit_all_gathers: true
68# fsdp_sync_module_states: true
69# fsdp_offload_params: true
70# fsdp_use_orig_params: false
71# fsdp_cpu_ram_efficient_loading: true
72# fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
73# fsdp_transformer_layer_cls_to_wrap: Qwen3DecoderLayer
74# fsdp_state_dict_type: FULL_STATE_DICT
75# fsdp_sharding_strategy: FULL_SHARD
76special_tokens:
77
78# save_first_step: true # uncomment this to validate checkpoint saving works with your config