From e87ec0793787e44ced23813144f0b84a2699b905 Mon Sep 17 00:00:00 2001 From: Douglas Hanley Date: Wed, 24 Apr 2024 08:02:49 -0500 Subject: [PATCH] fix argument name, move with ctx funcs --- llama.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/llama.h b/llama.h index 6b348ac38..cfc2ee873 100644 --- a/llama.h +++ b/llama.h @@ -390,9 +390,10 @@ extern "C" { LLAMA_API uint32_t llama_n_ubatch (const struct llama_context * ctx); LLAMA_API uint32_t llama_n_seq_max (const struct llama_context * ctx); + LLAMA_API enum llama_pooling_type llama_pooling_type(const struct llama_context * ctx); + LLAMA_API enum llama_vocab_type llama_vocab_type (const struct llama_model * model); LLAMA_API enum llama_rope_type llama_rope_type (const struct llama_model * model); - LLAMA_API enum llama_pooling_type llama_pooling_type(const struct llama_context * model); LLAMA_API int32_t llama_n_vocab (const struct llama_model * model); LLAMA_API int32_t llama_n_ctx_train(const struct llama_model * model);