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
from vllm.model_executor.layers.linear import (
ColumnParallelLinear,
MergedColumnParallelLinear,
QKVParallelLinear,
ReplicatedLinear,
RowParallelLinear,
)
......@@ -61,7 +60,7 @@ class KimiMLP(nn.Module):
hidden_size: int,
intermediate_size: int,
hidden_act: str,
quant_config: QKVParallelLinear | None = None,
quant_config: QuantizationConfig | None = None,
reduce_results: bool = True,
prefix: str = "",
) -> 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