Clean hidden_states_before_norm (#15485)

2025-12-20 09:38:46 +08:00
parent 3c116d5e5a
commit b9ebf0ed63
4 changed files with 29 additions and 32 deletions
@@ -1217,9 +1217,6 @@ class ScheduleBatch(ScheduleBatchDisaggregationDecodeMixin):
    # Diffusion LLM
    dllm_config: Optional[DllmConfig] = None

-    # For hidden states before normal
-    return_hidden_states_before_norm: bool = False
-
    @classmethod
    def init_new(
        cls,
@@ -2104,7 +2101,6 @@ class ScheduleBatch(ScheduleBatchDisaggregationDecodeMixin):
            dllm_config=self.dllm_config,
            reqs=self.reqs,
            has_grammar=self.has_grammar,
-            return_hidden_states_before_norm=self.return_hidden_states_before_norm,
            mamba_track_indices=self.mamba_track_indices,
            mamba_track_mask=self.mamba_track_mask,
            mamba_track_seqlens=self.mamba_track_seqlens,