|
@@ -6317,7 +6317,7 @@ struct ggml_tensor * ggml_alibi(
|
|
|
|
|
|
|
|
ggml_scratch_save(ctx);
|
|
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)[0] = n_past;
|
|
|
((int32_t *) b->data)[1] = n_head;
|
|
((int32_t *) b->data)[1] = n_head;
|