Skip to content

GitLab

Unverified Commit bc306fe5 authored by ZiTian Zhao's avatar ZiTian Zhao Committed by GitHub
Browse files

fix incorrect type annotation in KimiMLP (#27885) 


Signed-off-by: default avatarzitian.zhao <zitian.zhao@tencentmusic.com>
parent 103a468b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
vllm/model_executor/models/kimi_linear.py
View file @ bc306fe5
...@@ -22,7 +22,6 @@ from vllm.model_executor.layers.layernorm import RMSNorm ...@@ -22,7 +22,6 @@ from vllm.model_executor.layers.layernorm import RMSNorm
from vllm.model_executor.layers.linear import ( from vllm.model_executor.layers.linear import (
ColumnParallelLinear, ColumnParallelLinear,
MergedColumnParallelLinear, MergedColumnParallelLinear,
QKVParallelLinear,
ReplicatedLinear, ReplicatedLinear,
RowParallelLinear, RowParallelLinear,
) )
...@@ -61,7 +60,7 @@ class KimiMLP(nn.Module): ...@@ -61,7 +60,7 @@ class KimiMLP(nn.Module):
hidden_size: int, hidden_size: int,
intermediate_size: int, intermediate_size: int,
hidden_act: str, hidden_act: str,
quant_config: QKVParallelLinear | None = None, quant_config: QuantizationConfig | None = None,
reduce_results: bool = True, reduce_results: bool = True,
prefix: str = "", prefix: str = "",
) -> None: ) -> None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment