diff --git a/vllm/transformers_utils/config.py b/vllm/transformers_utils/config.py index f0f4074556cd7..b20c4139c398f 100644 --- a/vllm/transformers_utils/config.py +++ b/vllm/transformers_utils/config.py @@ -277,10 +277,10 @@ def get_pooling_config(model, revision='main'): pooling = next((item for item in modules_dict if item["type"] == "sentence_transformers.models.Pooling"), None) - normalize = next((item for item in modules_dict if + normalize = bool(next((item for item in modules_dict if item["type"] == "sentence_transformers.models.Normalize"), - False) + False)) if pooling: