| 1231 | }; |
| 1232 | |
| 1233 | static struct train_params get_default_train_params() { |
| 1234 | struct train_params params; |
| 1235 | params.common = get_default_train_params_common(); |
| 1236 | params.fn_model_base = ""; |
| 1237 | params.fn_lora_out = "ggml-lora-ITERATION-f32.gguf"; |
| 1238 | |
| 1239 | params.only_write_lora = false; |
| 1240 | |
| 1241 | params.f_norm_rms_eps = 1e-5f; |
| 1242 | params.rope_freq_base = 10000.0f; |
| 1243 | params.rope_freq_scale = 1.0f; |
| 1244 | |
| 1245 | params.custom_f_norm_rms_eps = false; |
| 1246 | params.custom_rope_freq_base = false; |
| 1247 | params.custom_rope_freq_scale = false; |
| 1248 | |
| 1249 | params.lora_r = 4; |
| 1250 | params.lora_alpha = 4; |
| 1251 | params.custom_lora_alpha = false; |
| 1252 | |
| 1253 | params.n_rank_attention_norm = 1; |
| 1254 | params.n_rank_wq = 4; |
| 1255 | params.n_rank_wk = 4; |
| 1256 | params.n_rank_wv = 4; |
| 1257 | params.n_rank_wo = 4; |
| 1258 | params.n_rank_ffn_norm = 1; |
| 1259 | params.n_rank_w1 = 4; |
| 1260 | params.n_rank_w2 = 4; |
| 1261 | params.n_rank_w3 = 4; |
| 1262 | params.n_rank_tok_embeddings = 4; |
| 1263 | params.n_rank_norm = 1; |
| 1264 | params.n_rank_output = 4; |
| 1265 | |
| 1266 | params.custom_n_rank_attention_norm = false; |
| 1267 | params.custom_n_rank_wq = false; |
| 1268 | params.custom_n_rank_wk = false; |
| 1269 | params.custom_n_rank_wv = false; |
| 1270 | params.custom_n_rank_wo = false; |
| 1271 | params.custom_n_rank_ffn_norm = false; |
| 1272 | params.custom_n_rank_w1 = false; |
| 1273 | params.custom_n_rank_w2 = false; |
| 1274 | params.custom_n_rank_w3 = false; |
| 1275 | params.custom_n_rank_tok_embeddings = false; |
| 1276 | params.custom_n_rank_norm = false; |
| 1277 | params.custom_n_rank_output = false; |
| 1278 | |
| 1279 | return params; |
| 1280 | } |
| 1281 | |
| 1282 | static void train_print_usage(int argc, char ** argv, const struct train_params * params) { |
| 1283 | fprintf(stderr, "usage: %s [options]\n", argv[0]); |
no test coverage detected