Views
No views yet
0.8.0.dev01base_model: unsloth/phi-4
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8
9# User Liger
10plugins:
11 - axolotl.integrations.liger.LigerPlugin
12liger_rope: true
13liger_rms_norm: true
14liger_glu_activation: true
15liger_fused_linear_cross_entropy: true
16
17chat_template: tokenizer_default
18datasets:
19 - path: shisa-ai/shisa-v2-best-of-n-athenev2-tulu70b-llama33-only-no-sysprompt
20 type: chat_template
21 field_messages: conversations
22 message_field_role: from
23 message_field_content: value
24 - path: shisa-ai/shisa-v2-roleplaying-sft
25 type: chat_template
26 field_messages: conversations
27 message_property_mappings:
28 role: role
29 content: content
30 roles:
31 system:
32 - system
33 assistant:
34 - gpt
35 - model
36 - assistant
37 user:
38 - human
39 - user
40 roles_to_train: ["assistant"]
41 - path: shisa-ai/translation_expanded_master_set_filtered
42 split: train[:25%]
43 type: chat_template
44 field_messages: conversations
45 message_property_mappings:
46 role: role
47 content: content
48 roles:
49 system:
50 - system
51 assistant:
52 - gpt
53 - model
54 - assistant
55 user:
56 - human
57 - user
58 roles_to_train: ["assistant"]
59 - path: shisa-ai/rewild-set-deepseek-subset
60 split: train[:25%]
61 type: chat_template
62 field_messages: conversations
63 message_property_mappings:
64 role: role
65 content: content
66 roles:
67 system:
68 - system
69 assistant:
70 - gpt
71 - model
72 - assistant
73 user:
74 - human
75 - user
76 roles_to_train: ["assistant"]
77 - path: shisa-ai/magpie-ultra-set
78 split: train[:8%]
79 type: chat_template
80 field_messages: conversations
81 message_property_mappings:
82 role: role
83 content: content
84 roles:
85 system:
86 - system
87 assistant:
88 - gpt
89 - model
90 - assistant
91 user:
92 - human
93 - user
94 roles_to_train: ["assistant"]
95 - path: shisa-ai/magpie-advanced-questions-set
96 split: train[:8%]
97 type: chat_template
98 field_messages: conversations
99 message_property_mappings:
100 role: role
101 content: content
102 roles:
103 system:
104 - system
105 assistant:
106 - gpt
107 - model
108 - assistant
109 user:
110 - human
111 - user
112 roles_to_train: ["assistant"]
113 - path: shisa-ai/japan-magpie-set
114 split: train
115 type: chat_template
116 field_messages: conversations
117 message_property_mappings:
118 role: role
119 content: content
120 roles:
121 system:
122 - system
123 assistant:
124 - gpt
125 - model
126 - assistant
127 user:
128 - human
129 - user
130 roles_to_train: ["assistant"]
131
132dataset_prepared_path: last_run_prepared
133val_set_size: 0.05
134output_dir: ./outputs/ablation-153-finalsft-shisa-v2-unphi-4-14b
135
136sequence_len: 8192
137sample_packing: true
138pad_to_sequence_len: true
139
140# marginal difference
141neftune_noise_alpha: 5
142
143use_wandb: true
144wandb_project: shisa-v2
145wandb_entity: augmxnt
146wandb_name: ablation-153-finalsft-shisa-v2-unphi-4-14b
147
148gradient_accumulation_steps: 2
149micro_batch_size: 4
150num_epochs: 3
151optimizer: paged_adamw_8bit
152lr_scheduler: linear
153learning_rate: 7.5e-6
154
155train_on_inputs: false
156group_by_length: false
157bf16: auto
158fp16:
159tf32: false
160
161gradient_checkpointing: true
162gradient_checkpointing_kwargs:
163 use_reentrant: false
164early_stopping_patience:
165resume_from_checkpoint:
166logging_steps: 1
167xformers_attention:
168flash_attention: true
169
170warmup_steps: 100
171evals_per_epoch: 2
172eval_table_size:
173saves_per_epoch: 0
174save_total_limit: 1 # Only store a single checkpoint
175debug:
176deepspeed: zero3_bf16.json
177weight_decay: 0.0001
178fsdp:
179fsdp_config:
180special_tokens:
181 # pad_token: "<|dummy_87|>"
182| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.7096 | 0.0020 | 1 | 0.7202 |
| 0.5031 | 0.5 | 246 | 0.5037 |
| 0.4589 | 1.0 | 492 | 0.4841 |
| 0.4245 | 1.5 | 738 | 0.4780 |
| 0.4423 | 2.0 | 984 | 0.4719 |
| 0.3957 | 2.5 | 1230 | 0.4754 |
| 0.3923 | 3.0 | 1476 | 0.4735 |