better match for deepseek

InternLM · Mar 11, 2024 · b5e727e · b5e727e
1 parent c7d83fd
commit b5e727e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lmdeploy/model.py b/lmdeploy/model.py
@@ -826,7 +826,8 @@ def match(cls, model_path: str) -> Optional[str]:
         Args:
             model_path (str): the model path used for matching.
         """
-        if 'deepseek' in model_path.lower():
+        path = model_path.lower()
+        if 'deepseek' in path and 'chat' in path:
             return 'deepseek'