Skip to content

Commit

Permalink
chore: case-insensitive model_management.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Anush008 authored Mar 22, 2024
1 parent c651b2b commit 379f25b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fastembed/common/model_management.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ def _get_model_description(cls, model_name: str) -> Dict[str, Any]:
Dict[str, Any]: The model description.
"""
for model in cls.list_supported_models():
if model_name == model["model"]:
if model_name.lower() == model["model"].lower():
return model

raise ValueError(f"Model {model_name} is not supported in {cls.__name__}.")
Expand Down

0 comments on commit 379f25b

Please sign in to comment.