Use uint32_t for seed
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
This commit is contained in:
parent
5a16205274
commit
d7d454f227
1 changed files with 1 additions and 1 deletions
2
llama.h
2
llama.h
|
@ -198,7 +198,7 @@ extern "C" {
|
||||||
LLAMA_API int llama_get_kv_cache_token_count(const struct llama_context * ctx);
|
LLAMA_API int llama_get_kv_cache_token_count(const struct llama_context * ctx);
|
||||||
|
|
||||||
// Sets the current rng seed.
|
// Sets the current rng seed.
|
||||||
LLAMA_API void llama_set_rng_seed(struct llama_context * ctx, unsigned int seed);
|
LLAMA_API void llama_set_rng_seed(struct llama_context * ctx, uint32_t seed);
|
||||||
|
|
||||||
// Returns the maximum size in bytes of the state (rng, logits, embedding
|
// Returns the maximum size in bytes of the state (rng, logits, embedding
|
||||||
// and kv_cache) - will often be smaller after compacting tokens
|
// and kv_cache) - will often be smaller after compacting tokens
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue