Просмотр исходного кода

llama : correct rms norm for llama 4 (#12882)

Xuan-Son Nguyen 9 месяцев назад
Родитель
Сommit
8b91d5355a
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      src/llama-model.cpp

+ 2 - 2
src/llama-model.cpp

@@ -4440,8 +4440,8 @@ struct llm_build_llama : public llm_graph_context {
 
                 if (arch == LLM_ARCH_LLAMA4 && use_rope && hparams.use_kq_norm) {
                     // Llama4TextL2Norm
-                    Qcur = ggml_rms_norm(ctx0, Qcur, 1e-6);
-                    Kcur = ggml_rms_norm(ctx0, Kcur, 1e-6);
+                    Qcur = ggml_rms_norm(ctx0, Qcur, hparams.f_norm_rms_eps);
+                    Kcur = ggml_rms_norm(ctx0, Kcur, hparams.f_norm_rms_eps);
                     cb(Qcur, "Qcur_normed", il);
                     cb(Kcur, "Kcur_normed", il);
                 }