Unverified Commit 739daa63 authored by Liangsheng Yin's avatar Liangsheng Yin Committed by GitHub
Browse files

Adjust logits metada init for target verify (#11467)

parent d957177a
......@@ -393,7 +393,7 @@ class ForwardBatch:
ret.positions = ret.spec_info.positions
# Init position information
if ret.forward_mode.is_decode():
if ret.forward_mode.is_decode() or ret.forward_mode.is_target_verify():
if ret.positions is None:
ret.positions = clamp_position(batch.seq_lens)
else:
......
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