Update main.cpp

This commit is contained in:
rabidcopy 2023-03-22 11:43:57 -05:00 committed by GitHub
parent 879da33ab4
commit e590787ab3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

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