diff options
author | Georgi Gerganov <ggerganov@gmail.com> | 2023-05-29 19:30:49 +0300 |
---|---|---|
committer | Georgi Gerganov <ggerganov@gmail.com> | 2023-05-29 19:30:49 +0300 |
commit | 5d1830b99dfd85bb6279adb4dd94aa444afd5b5e (patch) | |
tree | 18ef2394ef160bab6c02d5c11269ed347e5496be | |
parent | 248367605ead6fb7c36d2bfb1ebd8f00a23f7c71 (diff) |
ggml : fix bug in ggml_alibi
-rw-r--r-- | ggml.c | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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; |