Fix template name of MiniCPM-V (#6620)
* fix template name * tiny fix Former-commit-id: 94dea52cef709a7e6f1cdc0b78e83e0422bd65d3
This commit is contained in:
@@ -254,7 +254,7 @@ class BasePlugin:
|
||||
return {}
|
||||
|
||||
|
||||
class CpmOPlugin(BasePlugin):
|
||||
class CpmVPlugin(BasePlugin):
|
||||
@override
|
||||
def process_messages(
|
||||
self,
|
||||
@@ -949,7 +949,7 @@ class MllamaPlugin(BasePlugin):
|
||||
|
||||
PLUGINS = {
|
||||
"base": BasePlugin,
|
||||
"cpm_o": CpmOPlugin,
|
||||
"cpm_v": CpmVPlugin,
|
||||
"llava": LlavaPlugin,
|
||||
"llava_next": LlavaNextPlugin,
|
||||
"llava_next_video": LlavaNextVideoPlugin,
|
||||
|
||||
@@ -567,12 +567,12 @@ _register_template(
|
||||
|
||||
|
||||
_register_template(
|
||||
name="cpm_o",
|
||||
name="cpm_v",
|
||||
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
||||
format_assistant=StringFormatter(slots=["{{content}}<|im_end|>\n"]),
|
||||
format_system=StringFormatter(slots=["<|im_start|>system\n{{content}}<|im_end|>\n"]),
|
||||
stop_words=["<|im_end|>"],
|
||||
mm_plugin=get_mm_plugin(name="cpm_o", image_token="<image>", video_token="<video>"),
|
||||
mm_plugin=get_mm_plugin(name="cpm_v", image_token="<image>", video_token="<video>"),
|
||||
)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user