Skip to content

Commit 334515f

Browse files
committed
llama : correct llama_set_warmup() description
1 parent 2121335 commit 334515f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

include/llama.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -946,7 +946,7 @@ extern "C" {
946946
LLAMA_API void llama_set_causal_attn(struct llama_context * ctx, bool causal_attn);
947947

948948
// Set whether the model is in warmup mode or not
949-
// If true, all model tensors are activated during llama_encode() to load and cache their weights.
949+
// If true, all model tensors are activated during llama_decode() to load and cache their weights.
950950
LLAMA_API void llama_set_warmup(struct llama_context * ctx, bool warmup);
951951

952952
// Set abort callback

0 commit comments

Comments
 (0)