fix: adopting review suggestions

akai-shuuichi · Nov 13, 2024 · 62044b8 · 62044b8
1 parent d49b6dd
commit 62044b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lmdeploy/model.py b/lmdeploy/model.py
@@ -1035,7 +1035,7 @@ 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():
+        if 'qwen2.5' in model_path.lower() or 'qwen2_5' in model_path.lower():
             return 'qwen2d5'