minor cleanup

Signed-off-by: Thomas Parnell <[email protected]>

Author: tdoublep

vllm/model_executor/models/minimax_text_01.py

@@ -578,12 +578,13 @@ def _forward(self, hidden_states: torch.Tensor, positions: torch.Tensor,
                 hidden = self._decode_infer(q, k, v, kv_cache,
                                             state_indices_tensor,
                                             attn_metadata)
+
         hidden = self.norm._forward(hidden)
         gate, _ = self.output_gate(hidden_states[:num_actual_tokens])
         hidden = F.sigmoid(gate) * hidden
         hidden = hidden.to(hidden_states.dtype)
-        output, _ = self.out_proj(hidden)
-        return output[:num_actual_tokens]
+        hidden, _ = self.out_proj(hidden)
+        return hidden[:num_actual_tokens]
 
 
 class MiniMaxText01Attention(nn.Module):