diff --git a/vllm/config.py b/vllm/config.py index 0540dc4170584..ed0192b1ca62b 100644 --- a/vllm/config.py +++ b/vllm/config.py @@ -433,8 +433,9 @@ def get_pooling_config( pooling_type = self.get_pooling_type( pooling_config["pooling_type"]) normalize = pooling_config["normalize"] - pooling_config = PoolingConfig(pooling_type=pooling_type, - normalize=normalize) + pooling_config = PoolingConfig( + pooling_type=pooling_type, + normalize=normalize) # type: ignore if pooling_type_arg is not None: pooling_config.pooling_type = self.get_pooling_type( pooling_type_arg)