diff --git a/model_adapter.cpp b/model_adapter.cpp index 5612da902..6864e0d37 100644 --- a/model_adapter.cpp +++ b/model_adapter.cpp @@ -126,7 +126,7 @@ void print_tok_vec(std::vector &embd) fileformat = FileFormat::GPTJ_3; //quantized format cannot be legacy type } } - else if(vocabsiz==50257) + else if(vocabsiz==50257 || (vocabsiz>=49152&&vocabsiz<=49157)) //49152-6 is starcoder { fileformat = FileFormat::GPT2_1; uint32_t temp;