瀏覽代碼

model: fix LFM2_MOE missing tensors (#18132)

Tarek Dakhran 1 月之前
父節點
當前提交
982060fadc
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      src/llama-arch.cpp

+ 1 - 1
src/llama-arch.cpp

@@ -2055,7 +2055,7 @@ static std::set<llm_tensor> llm_get_tensor_names(llm_arch arch) {
                 LLM_TENSOR_SHORTCONV_INPROJ,
                 LLM_TENSOR_SHORTCONV_OUTPROJ,
                 LLM_TENSOR_TOKEN_EMBD,
-                LLM_TENSOR_OUTPUT_NORM,
+                LLM_TENSOR_OUTPUT_NORM_LFM2,
                 LLM_TENSOR_FFN_GATE_INP,
                 LLM_TENSOR_FFN_GATE_EXPS,
                 LLM_TENSOR_FFN_DOWN_EXPS,