added support for starcoder, which is basically gpt2
This commit is contained in:
parent
6d7749c98f
commit
01a0f206df
1 changed files with 1 additions and 1 deletions
|
@ -126,7 +126,7 @@ void print_tok_vec(std::vector<float> &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;
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue