From ae73f4b0c0270f898b6de981705cbe13c7d5ce8a Mon Sep 17 00:00:00 2001 From: Flavia Beo Date: Mon, 28 Oct 2024 18:33:21 -0300 Subject: [PATCH] Format linting Signed-off-by: Flavia Beo --- vllm/config.py | 3 +-- vllm/transformers_utils/config.py | 4 ++-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/vllm/config.py b/vllm/config.py index 01c4a7af0fc4d..e06d498a70e78 100644 --- a/vllm/config.py +++ b/vllm/config.py @@ -439,8 +439,7 @@ def get_pooling_config( if not pooling_type_arg and not normalize_arg: return PoolingConfig( pooling_type=self.get_pooling_type(pooling_type), - normalize=normalize - ) + normalize=normalize) if pooling_type_arg: pooling_type = pooling_type_arg if normalize_arg: diff --git a/vllm/transformers_utils/config.py b/vllm/transformers_utils/config.py index e2adbb5a3da05..8c788f7918040 100644 --- a/vllm/transformers_utils/config.py +++ b/vllm/transformers_utils/config.py @@ -317,7 +317,7 @@ def get_pooling_config(model, revision='main', token: Optional[str] = None): token) pooling_type_name = next( (item for item, val in pooling_dict.items() if val is True), None) - + pooling_type_name = get_pooling_config_name(pooling_type_name) return {"pooling_type": pooling_type_name, "normalize": normalize} @@ -336,7 +336,7 @@ def get_pooling_config_name(pooling_name: str): pooling_name = "last" pooling_type_name = pooling_name.upper() - + return pooling_type_name