Skip to content

Commit b3031d2

Browse files
authored
fix mask (#5385)
1 parent 36b6506 commit b3031d2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fastdeploy/spec_decode/mtp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -757,7 +757,7 @@ def _propose(self, step_use_cudagraph: bool = False, is_dummy_run: bool = False)
757757
self.model_inputs["is_block_step"],
758758
self.model_inputs["decode_states"],
759759
self.model_inputs["mask_rollback"],
760-
)[0]
760+
)
761761
self.model_inputs["attn_mask_offsets"].copy_(attn_mask_offsets, False)
762762

763763
# Initialize forward meta data

0 commit comments

Comments
 (0)