Make normalize variable return bool value

Signed-off-by: Flavia Beo <[email protected]>
flaviabeo · Oct 21, 2024 · 0546a0c · 0546a0c
1 parent 0d6123a
commit 0546a0c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: