From ddc21a01f5e1edc9171723dc59f817f9c1e48c4f Mon Sep 17 00:00:00 2001 From: AllentDan Date: Mon, 2 Dec 2024 17:11:06 +0800 Subject: [PATCH] fix template, using original internlm2 template --- lmdeploy/model.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/lmdeploy/model.py b/lmdeploy/model.py index 3406fb381f..a4355ea131 100644 --- 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]: