From 879da33ab4ad4db52ce99e1eeb5394c0839bd878 Mon Sep 17 00:00:00 2001 From: rabidcopy Date: Wed, 22 Mar 2023 11:41:19 -0500 Subject: [PATCH] Update main.cpp --- main.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.cpp b/main.cpp index b64e9d13f..bd35270e0 100644 --- a/main.cpp +++ b/main.cpp @@ -250,9 +250,9 @@ int main(int argc, char ** argv) { } // 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()) { - 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