Unverified Commit 50af4cdc authored by Chaitanya Malakala's avatar Chaitanya Malakala Committed by GitHub
Browse files

fix: return None for hidden stop tokens to prevent EOS in output (#5238)


Signed-off-by: default avatarChaitanya Malakala <106723863+chay1045@users.noreply.github.com>
parent 66dfc494
......@@ -456,7 +456,7 @@ impl Decoder {
// check for hidden stop tokens - eos takes precedence
if self.hidden_stop_ids.contains(&token_id) {
return Ok(StepResult::with_stop_trigger(
token,
None,
StopTrigger::HiddenStopTokenDetected(token_id),
));
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment