Skip to content

Commit 6a7eea7

Browse files
committed
fix compilation error
1 parent c3ecff5 commit 6a7eea7

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

include/llama.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -944,7 +944,7 @@ extern "C" {
944944
LLAMA_API void llama_set_embeddings(struct llama_context * ctx, bool embeddings);
945945

946946
// Set the attention type
947-
LLAMA_API void llama_set_attn_type(struct llama_context * ctx, llama_attention_type type);
947+
LLAMA_API void llama_set_attn_type(struct llama_context * ctx, enum llama_attention_type type);
948948

949949
// Set whether to use causal attention or not
950950
// - true: the model will only attend to the past tokens, alias of LLAMA_ATTENTION_TYPE_CAUSAL

src/llama-context.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2410,7 +2410,7 @@ void llama_set_embeddings(llama_context * ctx, bool embeddings) {
24102410
ctx->set_embeddings(embeddings);
24112411
}
24122412

2413-
void llama_set_attn_type(llama_context * ctx, llama_attention_type type) {
2413+
void llama_set_attn_type(llama_context * ctx, enum llama_attention_type type) {
24142414
ctx->set_attn_type(type);
24152415
}
24162416

0 commit comments

Comments
 (0)