match is chat

InternLM · Mar 6, 2024 · b459083 · b459083
1 parent 34a36ce
commit b459083
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lmdeploy/model.py b/lmdeploy/model.py
@@ -367,7 +367,7 @@ def match(cls, model_path: str) -> Optional[str]:
             model_path (str): the model path used for matching.
         """
         path = model_path.lower()
-        if 'internlm2' in path and 'chat' in path:
+        if 'internlm2' in path and ('chat' in path or 'math' in path):
             return 'internlm2-chat'