Fix token generation in bilingual models (non-English outputs) (#1668)

Co-authored-by: Guoqing Bao <guoqing.bao@enflame-tech.com>
This commit is contained in:
Guoqing Bao
2024-02-06 19:03:53 +08:00
committed by GitHub
parent b545f54a19
commit 678f64dd27
2 changed files with 2 additions and 1 deletions

View File

@ -104,6 +104,7 @@ impl TextGeneration {
break;
}
if let Some(t) = self.tokenizer.next_token(next_token)? {
let t = t.replace("<|im_end|>", "\n");
print!("{t}");
std::io::stdout().flush()?;
}