chore: case insensitive check

qdrant · Mar 20, 2024 · 36ec049 · 36ec049
1 parent dc7f1ac
commit 36ec049
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fastembed/sparse/sparse_text_embedding.py b/fastembed/sparse/sparse_text_embedding.py
@@ -48,7 +48,7 @@ def __init__(
 
         for EMBEDDING_MODEL_TYPE in self.EMBEDDINGS_REGISTRY:
             supported_models = EMBEDDING_MODEL_TYPE.list_supported_models()
-            if any(model_name == model["model"] for model in supported_models):
+            if any(model_name.lower() == model["model"].lower() for model in supported_models):
                 self.model = EMBEDDING_MODEL_TYPE(model_name, cache_dir, threads, **kwargs)
                 return
 

diff --git a/fastembed/text/text_embedding.py b/fastembed/text/text_embedding.py
@@ -55,7 +55,7 @@ def __init__(
 
         for EMBEDDING_MODEL_TYPE in self.EMBEDDINGS_REGISTRY:
             supported_models = EMBEDDING_MODEL_TYPE.list_supported_models()
-            if any(model_name == model["model"] for model in supported_models):
+            if any(model_name.lower() == model["model"].lower() for model in supported_models):
                 self.model = EMBEDDING_MODEL_TYPE(model_name, cache_dir, threads, **kwargs)
                 return