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

More checks before assuming FIM tokens for Llama arch

Browse source
This commit is contained in:
Sigbjørn Skjæret 2024-05-30 16:33:06 +02:00 • committed by GitHub
parent db10f01310
commit 998d208e14
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama.cpp
View file

@ -4330,7 +4330,7 @@ static void llm_load_vocab(
[](unsigned char c){ return std::tolower(c); });
if (gen_name.find("code") != std::string::npos) {
if (model.arch == LLM_ARCH_LLAMA) {
if (model.arch == LLM_ARCH_LLAMA && gen_name.find("llama") != std::string::npos) {
vocab.special_prefix_id = 32007;
vocab.special_suffix_id = 32008;
vocab.special_middle_id = 32009;