vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added support for starcoder, which is basically gpt2

Browse source
This commit is contained in:
Concedo 2023-05-27 13:35:40 +08:00
parent 6d7749c98f
commit 01a0f206df
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
model_adapter.cpp
View file

@ -126,7 +126,7 @@ void print_tok_vec(std::vector<float> &embd)
fileformat = FileFormat::GPTJ_3; //quantized format cannot be legacy type 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; fileformat = FileFormat::GPT2_1;
uint32_t temp; uint32_t temp;