Another antiprompt fix

This commit is contained in:
beiller 2023-03-08 16:44:50 -05:00
parent 9425a21db5
commit ce7ebb3319
No known key found for this signature in database
GPG key ID: 5AC5D1B01D0E5D75

View file

@ -1004,7 +1004,8 @@ int main(int argc, char ** argv) {
buf[n_read+1] = 0;
}
std::vector<gpt_vocab::id> line_inp = ::llama_tokenize(vocab, buf, false);
std::vector<gpt_vocab::id> line_inp;
processor.Encode(buf, &antiprompt_inp);
embd_inp.insert(embd_inp.end(), line_inp.begin(), line_inp.end());
input_noecho = true; // do not echo this again