From 0546a0cd19cd94a5a145ceb291c2653ab1edaaeb Mon Sep 17 00:00:00 2001 From: Flavia Beo Date: Mon, 21 Oct 2024 13:27:00 -0300 Subject: [PATCH] Make normalize variable return bool value Signed-off-by: Flavia Beo --- vllm/transformers_utils/config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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: