Update llama.h

Added review comments

Co-authored-by: Pavol Rusnak <pavol@rusnak.io>
This commit is contained in:
Christian Falch 2023-04-01 18:46:37 +02:00 committed by GitHub
parent f411251bcf
commit 17f463a083
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -85,7 +85,7 @@ extern "C" {
// Returns the KV cache that will contain the context for the
// ongoing prediction with the model.
LLAMA_API uint8_t* llama_get_kv_cache(struct llama_context * ctx);
LLAMA_API const uint8_t * llama_get_kv_cache(struct llama_context * ctx);
// Returns the size of the KV cache
LLAMA_API size_t llama_get_kv_cache_size(struct llama_context * ctx);