fix template, using original internlm2 template

AllentDan · Dec 2, 2024 · ddc21a0 · ddc21a0
1 parent c7dc50b
commit ddc21a0
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/lmdeploy/model.py b/lmdeploy/model.py
@@ -603,10 +603,13 @@ def match(cls, model_path: str) -> Optional[str]:
 
 
 @MODELS.register_module(name='internvl2_5')
-class InternVL2_5(InternVL2InternLM2):
+class InternVL2_5(InternLM2Chat7B):
 
-    def __init__(self, separator='\n', **kwargs):
-        super().__init__(separator=separator, **kwargs)
+    def __init__(
+            self,
+            meta_instruction='你是书生·万象，英文名是InternVL，是由上海人工智能实验室、清华大学及多家合作单位联合开发的多模态大语言模型。',  # noqa
+            **kwargs):
+        super().__init__(meta_instruction=meta_instruction, **kwargs)
 
     @classmethod
     def match(cls, model_path: str) -> Optional[str]: