From: Georgi Gerganov <redacted>
Date: Mon, 29 May 2023 16:30:49 +0000 (+0300)
Subject: ggml : fix bug in ggml_alibi
X-Git-Tag: gguf-v0.4.0~701
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=5d1830b99dfd85bb6279adb4dd94aa444afd5b5e;p=pkg%2Fggml%2Fsources%2Fllama.cpp

ggml : fix bug in ggml_alibi
---

diff --git a/ggml.c b/ggml.c
index 07ea92a5..b4570eb5 100644
--- a/ggml.c
+++ b/ggml.c
@@ -6317,7 +6317,7 @@ struct ggml_tensor * ggml_alibi(
 
     ggml_scratch_save(ctx);
 
-    struct ggml_tensor * b = ggml_new_tensor_1d(ctx, GGML_TYPE_I32, 2);
+    struct ggml_tensor * b = ggml_new_tensor_1d(ctx, GGML_TYPE_I32, 3);
 
     ((int32_t *) b->data)[0] = n_past;
     ((int32_t *) b->data)[1] = n_head;