@@ -31,7 +31,7 @@ struct train_state * init_train_state() {
3131
3232 state->opt = new struct ggml_opt_context ;
3333 state->opt ->ctx = NULL ;
34- state->opt ->params = ggml_opt_default_params (GGML_OPT_ADAM );
34+ state->opt ->params = ggml_opt_default_params (GGML_OPT_TYPE_ADAM );
3535 state->opt ->params .graph_size = LLAMA_TRAIN_MAX_NODES;
3636 state->opt ->loss_after = 0 .0f ;
3737
@@ -556,7 +556,7 @@ void load_opt_context_gguf(struct gguf_context * fctx, struct ggml_context * f_g
556556 std::string opt_type;
557557 GGUF_GET_KEY (fctx, opt_type, gguf_get_val_str, GGUF_TYPE_STRING, true , LLM_KV_OPTIMIZER_TYPE);
558558 if (opt_type == LLM_KV_OPTIMIZER_TYPE_ADAM) {
559- opt->params .type = GGML_OPT_ADAM ;
559+ opt->params .type = GGML_OPT_TYPE_ADAM ;
560560
561561 GGUF_GET_KEY (fctx, opt->adam .fx_best , gguf_get_val_f32, GGUF_TYPE_FLOAT32, true , LLM_KV_OPTIMIZER_ADAM_BEST_LOSS);
562562 GGUF_GET_KEY (fctx, opt->adam .fx_prev , gguf_get_val_f32, GGUF_TYPE_FLOAT32, true , LLM_KV_OPTIMIZER_ADAM_PREVIOUS_LOSS);
@@ -568,7 +568,7 @@ void load_opt_context_gguf(struct gguf_context * fctx, struct ggml_context * f_g
568568 copy_tensor_by_name (opt->adam .v , f_ggml_ctx, LLM_TENSOR_OPTIMIZER_ADAM_SECOND_MOMENTS);
569569 copy_tensor_by_name (opt->adam .pf , f_ggml_ctx, LLM_TENSOR_OPTIMIZER_ADAM_PAST_LOSS_VALUES);
570570 } else if (opt_type == LLM_KV_OPTIMIZER_TYPE_LBFGS) {
571- opt->params .type = GGML_OPT_LBFGS ;
571+ opt->params .type = GGML_OPT_TYPE_LBFGS ;
572572
573573 GGUF_GET_KEY (fctx, opt->params .lbfgs .m , gguf_get_val_u32, GGUF_TYPE_UINT32, true , LLM_KV_OPTIMIZER_LBFGS_APPROX_HESSIAN_COUNT);
574574 GGUF_GET_KEY (fctx, opt->lbfgs .fx_best , gguf_get_val_f32, GGUF_TYPE_FLOAT32, true , LLM_KV_OPTIMIZER_LBFGS_BEST_LOSS);
@@ -603,7 +603,7 @@ void save_opt_context_gguf(struct gguf_context * fctx, struct ggml_opt_context *
603603 gguf_set_val_bool (fctx, LLM_KV_OPTIMIZER_JUST_INITIALIZED, opt->just_initialized );
604604
605605 switch (opt->params .type ) {
606- case GGML_OPT_ADAM :
606+ case GGML_OPT_TYPE_ADAM :
607607 {
608608 gguf_set_val_str (fctx, LLM_KV_OPTIMIZER_TYPE, LLM_KV_OPTIMIZER_TYPE_ADAM);
609609 gguf_set_val_f32 (fctx, LLM_KV_OPTIMIZER_ADAM_BEST_LOSS, opt->adam .fx_best );
@@ -622,7 +622,7 @@ void save_opt_context_gguf(struct gguf_context * fctx, struct ggml_opt_context *
622622 gguf_add_tensor (fctx, opt->adam .pf );
623623 }
624624 } break ;
625- case GGML_OPT_LBFGS :
625+ case GGML_OPT_TYPE_LBFGS :
626626 {
627627 gguf_set_val_str (fctx, LLM_KV_OPTIMIZER_TYPE, LLM_KV_OPTIMIZER_TYPE_LBFGS);
628628 gguf_set_val_u32 (fctx, LLM_KV_OPTIMIZER_LBFGS_APPROX_HESSIAN_COUNT, opt->params .lbfgs .m );
0 commit comments