Explorar o código

ggml : fix bug in ggml_alibi

Georgi Gerganov %!s(int64=2) %!d(string=hai) anos
pai
achega
5d1830b99d
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      ggml.c

+ 1 - 1
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;