Views
No views yet
0.14.01base_model: kakaocorp/kanana-1.5-8b-instruct-2505
2hf_cache_dir: ../../../../data5/models
3
4load_in_8bit: false
5load_in_4bit: true
6
7datasets:
8 - path: data_cap_50.jsonl
9 type: chat_template
10 split: train
11
12dataset_prepared_path: preprocess
13val_set_size: 0
14output_dir: ./outputs-no_unique-8b
15dataloader_num_workers: 32
16
17sequence_len: 5120
18sample_packing: false
19eval_sample_packing: false
20pad_to_sequence_len: false
21
22adapter: qlora
23lora_model_dir:
24lora_r: 32
25lora_alpha: 16
26lora_dropout: 0.05
27lora_target_linear: true
28
29plugins:
30 - axolotl.integrations.liger.LigerPlugin
31liger_rope: true
32liger_rms_norm: true
33liger_swiglu: true
34liger_fused_linear_cross_entropy: true
35
36wandb_project: TTP_SFT_LLM_RE
37wandb_entity:
38wandb_watch:
39wandb_name: ttp_sft_kanana-1.5_base_unique_remove-8b
40wandb_log_model:
41hub_model_id: CaffeineThief/ttp_sft_kanana-1.5_base_unique_remove-8b
42hub_private_repo: false
43
44gradient_accumulation_steps: 1
45micro_batch_size: 16
46num_epochs: 2
47optimizer: adamw_torch_fused
48lr_scheduler: cosine
49learning_rate: 2e-5
50
51bf16: auto
52tf32: false
53
54gradient_checkpointing: false
55resume_from_checkpoint:
56logging_steps: 1
57flash_attention: true
58
59warmup_ratio: 0.05
60weight_decay: 0.01
61# evals_per_epoch: 1
62saves_per_epoch: 0
63
64fsdp:
65 - full_shard
66 - auto_wrap
67fsdp_config:
68 fsdp_state_dict_type: FULL_STATE_DICT
69 fsdp_transformer_layer_cls_to_wrap: LlamaDecoderLayer
70 fsdp_activation_checkpointing: true