Skip to content

Commit 3862d95

Browse files
authored
rope
1 parent f5d0305 commit 3862d95

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/llama-model.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5961,7 +5961,7 @@ struct llm_build_bert : public llm_graph_context {
59615961
Vcur = ggml_reshape_3d(ctx0, Vcur, n_embd_head, n_head_kv, n_tokens);
59625962

59635963
// RoPE
5964-
if (model.arch == LLM_ARCH_NOMIC_BERT || model.arch == LLM_ARCH_NOMIC_BERT_MOE) {
5964+
if (model.arch == LLM_ARCH_NOMIC_BERT || model.arch == LLM_ARCH_NOMIC_BERT_MOE || model.arch == LLM_ARCH_JINA_BERT_V3) {
59655965
Qcur = ggml_rope_ext(
59665966
ctx0, Qcur, inp_pos, nullptr,
59675967
n_rot, rope_type, n_ctx_orig, freq_base, freq_scale,

0 commit comments

Comments
 (0)