Skip to content

Commit 269de86

Browse files
committed
llama : fix Gemma rope type (#5691)
1 parent c393733 commit 269de86

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

llama.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12089,7 +12089,6 @@ enum llama_rope_type llama_rope_type(const struct llama_model * model) {
1208912089
case LLM_ARCH_ORION:
1209012090
case LLM_ARCH_INTERNLM2:
1209112091
case LLM_ARCH_MINICPM:
12092-
case LLM_ARCH_GEMMA:
1209312092
return LLAMA_ROPE_TYPE_NORM;
1209412093

1209512094
// the pairs of head values are offset by n_rot/2
@@ -12101,6 +12100,7 @@ enum llama_rope_type llama_rope_type(const struct llama_model * model) {
1210112100
case LLM_ARCH_QWEN:
1210212101
case LLM_ARCH_QWEN2:
1210312102
case LLM_ARCH_PHI2:
12103+
case LLM_ARCH_GEMMA:
1210412104
return LLAMA_ROPE_TYPE_NEOX;
1210512105

1210612106
// all model arches should be listed explicitly here

0 commit comments

Comments
 (0)