llama : fix old glm4 models (#11670)

ggerganov · Feb 6, 2025 · 855cd07 · 855cd07
1 parent 8a59053
commit 855cd07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/llama.cpp b/src/llama.cpp
@@ -7217,7 +7217,7 @@ struct llm_build_context {
                 struct ggml_tensor * Qcur = nullptr;
                 struct ggml_tensor * Kcur = nullptr;
                 struct ggml_tensor * Vcur = nullptr;
-                if (model.type == LLM_TYPE_1_5B || model.type == LLM_TYPE_4B || model.type == LLM_TYPE_9B) {
+                if (model.layers[il].wqkv == nullptr) {
                     Qcur = llm_build_lora_mm(lctx, ctx0, model.layers[il].wq, cur);
                     if (model.layers[il].bq) {
                         Qcur = ggml_add(ctx0, Qcur, model.layers[il].bq);