millaguie/llama.cpp
1
0 Fork 0
Code Issues Pull Requests Actions 29 Packages Projects Releases Wiki Activity

llama : add llama_n_head() (#9512)

Browse Source
This commit is contained in:
Michael Podvitskiy authored and GitHub committed 2024-09-17 09:23:30 +03:00
1 parent 23e0d70bac
commit 37f3a3810e
2 files changed
+5
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

include/llama.h
+1
View File
@@ -441,6 +441,7 @@ extern "C" {
LLAMA_API int32_t llama_n_ctx_train(const struct llama_model * model);
LLAMA_API int32_t llama_n_embd (const struct llama_model * model);
LLAMA_API int32_t llama_n_layer (const struct llama_model * model);
LLAMA_API int32_t llama_n_head (const struct llama_model * model);
LLAMA_API const struct llama_model * llama_get_model(const struct llama_context * ctx);