From 62044b89d9e027a90324f0a4111bd4366613c505 Mon Sep 17 00:00:00 2001 From: gaozixiang Date: Wed, 13 Nov 2024 15:57:16 +0800 Subject: [PATCH] fix: adopting review suggestions --- lmdeploy/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lmdeploy/model.py b/lmdeploy/model.py index 7f6d69866c..13c89b71b4 100644 --- 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'