File tree Expand file tree Collapse file tree 2 files changed +5
-0
lines changed Expand file tree Collapse file tree 2 files changed +5
-0
lines changed Original file line number Diff line number Diff line change @@ -441,6 +441,7 @@ extern "C" {
441441 LLAMA_API int32_t llama_n_ctx_train (const struct llama_model * model);
442442 LLAMA_API int32_t llama_n_embd (const struct llama_model * model);
443443 LLAMA_API int32_t llama_n_layer (const struct llama_model * model);
444+ LLAMA_API int32_t llama_n_head (const struct llama_model * model);
444445
445446 LLAMA_API const struct llama_model * llama_get_model (const struct llama_context * ctx);
446447
Original file line number Diff line number Diff line change @@ -19047,6 +19047,10 @@ int32_t llama_n_layer(const struct llama_model * model) {
1904719047 return model->hparams.n_layer;
1904819048}
1904919049
19050+ int32_t llama_n_head(const struct llama_model * model) {
19051+ return model->hparams.n_head();
19052+ }
19053+
1905019054const struct llama_model * llama_get_model(const struct llama_context * ctx) {
1905119055 return &ctx->model;
1905219056}
You can’t perform that action at this time.
0 commit comments