diff --git a/ggml/src/ggml.c b/ggml/src/ggml.c index 1c43051e86..5fd4707735 100644 --- a/ggml/src/ggml.c +++ b/ggml/src/ggml.c @@ -6104,7 +6104,7 @@ struct ggml_tensor * ggml_opt_step_adamw( GGML_ASSERT(ggml_are_same_shape(a, m)); GGML_ASSERT(ggml_are_same_shape(a, v)); GGML_ASSERT(adamw_params->type == GGML_TYPE_F32); - GGML_ASSERT(ggml_nelements(adamw_params) == 7); + GGML_ASSERT(ggml_nelements(adamw_params) == 8); struct ggml_tensor * result = ggml_view_tensor(ctx, a);