File tree Expand file tree Collapse file tree 2 files changed +7
-4
lines changed Expand file tree Collapse file tree 2 files changed +7
-4
lines changed Original file line number Diff line number Diff line change @@ -154,6 +154,7 @@ const char * llm_type_name(llm_type type) {
154154        case  LLM_TYPE_1_7B:          return  " 1.7B" 
155155        case  LLM_TYPE_1_8B:          return  " 1.8B" 
156156        case  LLM_TYPE_2B:            return  " 2B" 
157+         case  LLM_TYPE_2_6B:          return  " 2.6B" 
157158        case  LLM_TYPE_2_8B:          return  " 2.8B" 
158159        case  LLM_TYPE_2_9B:          return  " 2.9B" 
159160        case  LLM_TYPE_3B:            return  " 3B" 
@@ -2089,10 +2090,11 @@ void llama_model::load_hparams(llama_model_loader & ml) {
20892090                for  (uint32_t  il = 0 ; il < hparams.n_layer ; ++il) {
20902091                    hparams.recurrent_layer_arr [il] = hparams.n_head_kv (il) == 0 ;
20912092                }
2092-                 switch  (hparams.n_embd ) {
2093-                     case  1024 : type = LLM_TYPE_350M; break ;
2094-                     case  1536 : type = LLM_TYPE_700M; break ;
2095-                     case  2048 : type = LLM_TYPE_1_2B; break ;
2093+                 switch  (hparams.n_ff ()) {
2094+                     case   4608 : type = LLM_TYPE_350M; break ;
2095+                     case   6912 : type = LLM_TYPE_700M; break ;
2096+                     case   8192 : type = LLM_TYPE_1_2B; break ;
2097+                     case  10752 : type = LLM_TYPE_2_6B; break ;
20962098                    default :   type = LLM_TYPE_UNKNOWN;
20972099                }
20982100            } break ;
Original file line number Diff line number Diff line change @@ -58,6 +58,7 @@ enum llm_type {
5858    LLM_TYPE_1_7B,
5959    LLM_TYPE_1_8B,
6060    LLM_TYPE_2B,
61+     LLM_TYPE_2_6B,
6162    LLM_TYPE_2_8B,
6263    LLM_TYPE_2_9B,
6364    LLM_TYPE_3B,
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments