From b45908397d72dc3f701e8bf462783d1068bba810 Mon Sep 17 00:00:00 2001
From: AllentDan <AllentDan@yeah.net>
Date: Wed, 6 Mar 2024 18:08:59 +0800
Subject: [PATCH] match is chat

---
 lmdeploy/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lmdeploy/model.py b/lmdeploy/model.py
index 43a7c0f7ed..b38bcf7e62 100644
--- 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'