fix kernel output extract (#5212)

2025-12-24 13:28:13 +08:00 · 2025-11-25 14:25:20 +08:00
parent a418d7b60b
commit e581b7d7d9
1 changed files with 1 additions and 1 deletions
--- 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