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

Update main.cpp

Browse source
This commit is contained in:
rabidcopy 2023-03-22 11:41:19 -05:00 • committed by GitHub
parent c4efdb22af
commit 879da33ab4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
main.cpp
View file

@ -250,9 +250,9 @@ int main(int argc, char ** argv) {
} }
// tokenize the first reverse prompt // tokenize the first reverse prompt
auto first_antiprompt = ::llama_tokenize(ctx,"",false); auto first_antiprompt = ::llama_tokenize(ctx, params.prompt, true);
if (!params.antiprompt.empty()) { if (!params.antiprompt.empty()) {
auto first_antiprompt = ::llama_tokenize(ctx, params.antiprompt.front(), false); auto first_antiprompt = ::llama_tokenize(ctx, params.antiprompt.front(), true);
} }
// enable interactive mode if reverse prompt is specified // enable interactive mode if reverse prompt is specified