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

main.cpp : insert bos if no tokens

Browse source
This commit is contained in:
klosax 2023-08-23 01:50:17 +02:00 • committed by GitHub
parent 777f42ba18
commit 9407c847a2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
examples/main/main.cpp
View file

@ -197,6 +197,11 @@ int main(int argc, char ** argv) {
embd_inp = session_tokens; embd_inp = session_tokens;
} }
// Should not run without any tokens
if(embd_inp.size()==0) {
embd_inp.push_back(llama_token_bos(ctx));
}
// Tokenize negative prompt // Tokenize negative prompt
std::vector<llama_token> guidance_inp; std::vector<llama_token> guidance_inp;
int guidance_offset = 0; int guidance_offset = 0;