Skip to content

Commit 4abde12

Browse files
authored
change beta fast for grok if default value
1 parent 6cf16aa commit 4abde12

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/llama-context.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,10 @@ llama_context::llama_context(
7070
cparams.yarn_ext_factor = rope_scaling_type == LLAMA_ROPE_SCALING_TYPE_YARN ? 1.0f : 0.0f;
7171
}
7272

73+
if (model.arch == LLM_ARCH_GROK && params.yarn_beta_fast == 32.0f) {
74+
cparams.yarn_beta_fast = 8.0f;
75+
}
76+
7377
cparams.yarn_attn_factor *= hparams.rope_attn_factor;
7478

7579
if (cparams.pooling_type == LLAMA_POOLING_TYPE_UNSPECIFIED) {

0 commit comments

Comments
 (0)