Fix internlm2 template

Author: xusenlin

api/adapter/template.py

@@ -721,11 +721,13 @@ def template(self) -> str:
 class InternLMTemplate(BaseTemplate):
 
     name = "internlm"
-    allow_models = ["internlm"]
     stop = {
         "strings": ["</s>", "<eoa>"],
     }
 
+    def match(self, name) -> bool:
+        return name.startswith("internlm") and not name.startswith("internlm2")
+
     @property
     def template(self) -> str:
         """ The output should look something like:
@@ -749,7 +751,6 @@ def template(self) -> str:
 class InternLM2Template(BaseTemplate):
 
     name = "internlm2"
-    allow_models = ["internlm2"]
     system_prompt = (
         "You are an AI assistant whose name is InternLM (书生·浦语).\n"
         "- InternLM (书生·浦语) is a conversational language model that is developed by Shanghai AI Laboratory (上海人工智能实验室). It is designed to be helpful, honest, and harmless.\n"
@@ -759,6 +760,9 @@ class InternLM2Template(BaseTemplate):
         "strings": ["</s>", "[UNUSED_TOKEN_145]"],
     }
 
+    def match(self, name) -> bool:
+        return name.startswith("internlm2")
+
     @property
     def template(self) -> str:
         return (

libs/langchain_llm/langchain_llm/adapters/template.py

@@ -721,11 +721,13 @@ def template(self) -> str:
 class InternLMTemplate(BaseTemplate):
 
     name = "internlm"
-    allow_models = ["internlm"]
     stop = {
         "strings": ["</s>", "<eoa>"],
     }
 
+    def match(self, name) -> bool:
+        return name.startswith("internlm") and not name.startswith("internlm2")
+
     @property
     def template(self) -> str:
         """ The output should look something like:
@@ -749,7 +751,6 @@ def template(self) -> str:
 class InternLM2Template(BaseTemplate):
 
     name = "internlm2"
-    allow_models = ["internlm2"]
     system_prompt = (
         "You are an AI assistant whose name is InternLM (书生·浦语).\n"
         "- InternLM (书生·浦语) is a conversational language model that is developed by Shanghai AI Laboratory (上海人工智能实验室). It is designed to be helpful, honest, and harmless.\n"
@@ -759,6 +760,9 @@ class InternLM2Template(BaseTemplate):
         "strings": ["</s>", "[UNUSED_TOKEN_145]"],
     }
 
+    def match(self, name) -> bool:
+        return name.startswith("internlm2")
+
     @property
     def template(self) -> str:
         return (