Don't display the endoftext token.

This commit is contained in:
Adam Treat 2023-04-09 01:22:12 -04:00
parent 02e13737f3
commit bd5e279621

View File

@ -740,6 +740,7 @@ void GPTJ::prompt(const std::string &prompt, std::function<bool(const std::strin
// add it to the context
embd.push_back(id);
if (id != 50256)
resp.push_back(id);
} else {
// if here, it means we are still processing the input prompt