fix: adopting review suggestions

akai-shuuichi · Nov 13, 2024 · 1ddb995 · 1ddb995
1 parent 62044b8
commit 1ddb995
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lmdeploy/model.py b/lmdeploy/model.py
@@ -1035,7 +1035,8 @@ def match(cls, model_path: str) -> Optional[str]:
         Args:
             model_path (str): the model path used for matching.
         """
-        if 'qwen2.5' in model_path.lower() or 'qwen2_5' in model_path.lower():
+        lower_path = model_path.lower()
+        if 'qwen2.5' in lower_path or 'qwen2_5' in lower_path:
             return 'qwen2d5'