Skip to content

Commit 5d1830b

Browse files
committed
ggml : fix bug in ggml_alibi
1 parent 2483676 commit 5d1830b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6317,7 +6317,7 @@ struct ggml_tensor * ggml_alibi(
63176317

63186318
ggml_scratch_save(ctx);
63196319

6320-
struct ggml_tensor * b = ggml_new_tensor_1d(ctx, GGML_TYPE_I32, 2);
6320+
struct ggml_tensor * b = ggml_new_tensor_1d(ctx, GGML_TYPE_I32, 3);
63216321

63226322
((int32_t *) b->data)[0] = n_past;
63236323
((int32_t *) b->data)[1] = n_head;

0 commit comments

Comments
 (0)