Skip to content

Commit c57d39e

Browse files
committed
Fix return type
Signed-off-by: Thomas Parnell <[email protected]>
1 parent bf57695 commit c57d39e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/model_executor/models/minimax_text_01.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -654,7 +654,7 @@ def __init__(
654654
return
655655

656656
def forward(self, hidden_states: torch.Tensor, positions: torch.Tensor,
657-
**kwargs) -> None:
657+
**kwargs) -> torch.Tensor:
658658
forward_context = get_forward_context()
659659
attn_metadata = forward_context.attn_metadata
660660
qkv, _ = self.qkv_proj(hidden_states)

0 commit comments

Comments
 (0)