Adjust logits metada init for target verify (#11467)
This commit is contained in:
@@ -393,7 +393,7 @@ class ForwardBatch:
|
|||||||
ret.positions = ret.spec_info.positions
|
ret.positions = ret.spec_info.positions
|
||||||
|
|
||||||
# Init position information
|
# 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:
|
if ret.positions is None:
|
||||||
ret.positions = clamp_position(batch.seq_lens)
|
ret.positions = clamp_position(batch.seq_lens)
|
||||||
else:
|
else:
|
||||||
|
|||||||
Reference in New Issue
Block a user