changed to use uint32_t instead of int

This commit is contained in:
marcus 2023-12-19 15:19:26 -08:00
parent d173ddac9e
commit 9b57750b69
2 changed files with 2 additions and 2 deletions

View file

@ -9536,7 +9536,7 @@ int llama_n_ctx(const struct llama_context * ctx) {
return ctx->cparams.n_ctx; return ctx->cparams.n_ctx;
} }
int llama_n_batch(const struct llama_context * ctx) { uint32_t llama_n_batch(const struct llama_context * ctx) {
return ctx->cparams.n_batch; return ctx->cparams.n_batch;
} }

View file

@ -315,7 +315,7 @@ extern "C" {
LLAMA_API const struct llama_model * llama_get_model(const struct llama_context * ctx); LLAMA_API const struct llama_model * llama_get_model(const struct llama_context * ctx);
LLAMA_API int llama_n_ctx (const struct llama_context * ctx); LLAMA_API int llama_n_ctx (const struct llama_context * ctx);
LLAMA_API int llama_n_batch (const struct llama_context * ctx); LLAMA_API uint32_t llama_n_batch (const struct llama_context * ctx);
LLAMA_API enum llama_vocab_type llama_vocab_type(const struct llama_model * model); LLAMA_API enum llama_vocab_type llama_vocab_type(const struct llama_model * model);