File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -8509,7 +8509,7 @@ static struct ggml_tensor * llm_build_kqv(
85098509        struct ggml_tensor * kq = ggml_mul_mat(ctx, k, q);
85108510        cb(kq, "kq", il);
85118511
8512-         if (model.arch == LLM_ARCH_PHI2 || model.arch == LLM_ARCH_PHI3 || model.arch == LLM_ARCH_GPTNEOX || model.arch == LLM_ARCH_QWEN2 || model.arch == LLM_ARCH_NEMOTRON) {
8512+         if (model.arch == LLM_ARCH_PHI2 || model.arch == LLM_ARCH_PHI3 || model.arch == LLM_ARCH_GPTNEOX || model.arch == LLM_ARCH_QWEN2 || model.arch == LLM_ARCH_NEMOTRON || model.arch == LLM_ARCH_CHATGLM ) {
85138513            // for this arch, we need to perform the KQ multiplication with F32 precision, otherwise we get NaNs
85148514            // ref: https://github.com/ggerganov/llama.cpp/pull/4490#issuecomment-1859055847
85158515            ggml_mul_mat_set_prec(kq, GGML_PREC_F32);
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments