Skip to content

Commit 6dd03a8

Browse files
BillHamShopbeiller
authored andcommitted
Another antiprompt fix
1 parent f67d8b0 commit 6dd03a8

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

main.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1004,7 +1004,8 @@ int main(int argc, char ** argv) {
10041004
buf[n_read+1] = 0;
10051005
}
10061006

1007-
std::vector<gpt_vocab::id> line_inp = ::llama_tokenize(vocab, buf, false);
1007+
std::vector<gpt_vocab::id> line_inp;
1008+
processor.Encode(buf, &antiprompt_inp);
10081009
embd_inp.insert(embd_inp.end(), line_inp.begin(), line_inp.end());
10091010

10101011
input_noecho = true; // do not echo this again

0 commit comments

Comments
 (0)