Skip to content

Commit f298a83

Browse files
committed
fix
1 parent 28471eb commit f298a83

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/llama.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -606,8 +606,8 @@ static struct ggml_tensor * llm_build_kqv(
606606
if (n_embd_head_v < n_embd_head_k) {
607607
cur = ggml_cont(ctx, cur);
608608
cur = ggml_cont(ctx, ggml_view_3d(ctx, cur, n_embd_head_v, n_head, n_tokens,
609-
ggml_row_size(cur->type, n_embd_head_v_out),
610-
ggml_row_size(cur->type, n_embd_head_v_out * n_head),
609+
cur->nb[1],
610+
cur->nb[2],
611611
0));
612612
}
613613

0 commit comments

Comments
 (0)