From 214942e1ae92f2d3ca7e2e8f93b3a5dd07ff72ad Mon Sep 17 00:00:00 2001 From: freeliuzc Date: Wed, 26 Nov 2025 16:48:42 +0800 Subject: [PATCH] fix kernel output extract (#5208) --- fastdeploy/spec_decode/mtp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastdeploy/spec_decode/mtp.py b/fastdeploy/spec_decode/mtp.py index 09cbb9566..321ef5a1b 100644 --- a/fastdeploy/spec_decode/mtp.py +++ b/fastdeploy/spec_decode/mtp.py @@ -799,7 +799,7 @@ class MTPProposer(Proposer): self.model_inputs["is_block_step"], self.model_inputs["decode_states"], self.model_inputs["mask_rollback"], - )[0] + ) self.model_inputs["attn_mask_offsets"].copy_(attn_mask_offsets, False) # Initialize forward meta data