tiny fix chain-style multi layer eagle comments (#22206)

2026-04-06 13:49:03 -07:00
parent 8f337682bd
commit 93f38fe410
1 changed files with 1 additions and 1 deletions
@@ -429,7 +429,7 @@ class MultiLayerEagleDraftExtendCudaGraphRunner:
                forward_batch,
            )

-            # Chain-style MTP: overwrite self.hidden_states with the draft model's
+            # Chain-style MTP: overwrite buffers.hidden_states with the draft model's
            # output (hidden_states_before_norm) so that assign_new_state_triton
            # propagates each MTP layer's own output to the next MTP layer,
            # rather than always feeding the target model's hidden states.