From e581b7d7d97fd21c619b6bd34cafb5eb72944ae6 Mon Sep 17 00:00:00 2001 From: freeliuzc Date: Tue, 25 Nov 2025 14:25:20 +0800 Subject: [PATCH] fix kernel output extract (#5212) --- 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 1403bd3d2..d7648c3e7 100644 --- a/fastdeploy/spec_decode/mtp.py +++ b/fastdeploy/spec_decode/mtp.py @@ -793,7 +793,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