fixed unbantokens not following EOS

This commit is contained in:
Concedo 2023-05-01 00:02:45 +08:00
parent 294a5d00b1
commit 25201233ca

View file

@ -674,6 +674,11 @@ generation_outputs gpttype_generate(const generation_inputs inputs, generation_o
if (file_format == FileFormat::GGML || file_format == FileFormat::GGHF || file_format == FileFormat::GGJT)
{
concat_output += llama_token_to_str(llama_ctx_v1, id);
if(unbanTokens && id==llama_token_eos())
{
printf("\n(EOS token triggered!)");
remaining_tokens = 0;
}
}
else
{