llama : add llama_n_head() (#9512)
This commit is contained in:
1 parent
23e0d70bac
commit
37f3a3810e
2 files changed
+5
No files matched your search
@@ -19047,6 +19047,10 @@ int32_t llama_n_layer(const struct llama_model * model) {
|
||||
return model->hparams.n_layer;
|
||||
}
|
||||
|
||||
int32_t llama_n_head(const struct llama_model * model) {
|
||||
return model->hparams.n_head();
|
||||
}
|
||||
|
||||
const struct llama_model * llama_get_model(const struct llama_context * ctx) {
|
||||
return &ctx->model;
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user