fix kernel output extract (#5208)

2025-12-24 13:28:13 +08:00 · 2025-11-26 16:48:42 +08:00
parent df427ba06d
commit 214942e1ae
1 changed files with 1 additions and 1 deletions
--- 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