Skip to content

Commit 3fec197

Browse files
committed
fixup falcon with new ggml
1 parent fa26740 commit 3fec197

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

implementations/falcon.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -461,8 +461,8 @@ bool falcon_eval(
461461
fused_qkv_row_nb, (n_embd + head_dim) * sizeof(float));
462462

463463
// using mode = 2 for neox mode
464-
Qcur = ggml_rope_inplace(ctx0, Qcur, n_past, head_dim, 2);
465-
Kcur = ggml_rope_inplace(ctx0, Kcur, n_past, head_dim, 2);
464+
Qcur = ggml_rope_inplace(ctx0, Qcur, n_past, head_dim, 2, 0);
465+
Kcur = ggml_rope_inplace(ctx0, Kcur, n_past, head_dim, 2, 0 );
466466

467467
// store key and value to memory
468468
{

0 commit comments

Comments
 (0)