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

Remove unnecessary use of std::stringstream.

Browse source
This commit is contained in:
Johnman 2023-03-20 16:31:10 +01:00
parent 6242d1ccd5
commit 12280807c3
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
main.cpp
View file

@ -12,7 +12,6 @@
#include <map> #include <map>
#include <string> #include <string>
#include <vector> #include <vector>
#include <sstream>
#if defined (__unix__) || (defined (__APPLE__) && defined (__MACH__)) #if defined (__unix__) || (defined (__APPLE__) && defined (__MACH__))
#include <signal.h> #include <signal.h>
@ -1023,11 +1022,10 @@ int main(int argc, char ** argv) {
// check if we should prompt the user for more // check if we should prompt the user for more
if (params.interactive && embd_inp.size() <= input_consumed) { if (params.interactive && embd_inp.size() <= input_consumed) {
// check for reverse prompt // check for reverse prompt
std::stringstream last_output_ss; std::string last_output;
for (auto id : last_n_tokens) { for (auto id : last_n_tokens) {
last_output_ss << vocab.id_to_token[id]; last_output += vocab.id_to_token[id];
} }
std::string last_output = last_output_ss.str();
// Check if each of the reverse prompts appears at the end of the output. // Check if each of the reverse prompts appears at the end of the output.
for (std::string antiprompt : params.antiprompt) { for (std::string antiprompt : params.antiprompt) {