update chat template for internlm2 base model (#1079)

InternLM · Jan 31, 2024 · f19ca25 · f19ca25
1 parent a41f565
commit f19ca25
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lmdeploy/model.py b/lmdeploy/model.py
@@ -322,7 +322,8 @@ def __init__(self, session_len=4096, capability='completion', **kwargs):
                          **kwargs)
 
 
-@MODELS.register_module(name=['internlm2-7b', 'internlm2-20b'])
+@MODELS.register_module(
+    name=['internlm2', 'internlm2-1_8b', 'internlm2-7b', 'internlm2-20b'])
 class InternLM2BaseModel7B(BaseModel):
     """Generation parameters of InternLM2-7B-Base model."""