根据代码规范修改了代码

Former-commit-id: fe5351980b42e0e38175b0da2401a61b3807fa7c
This commit is contained in:
liudan 2024-09-05 20:17:55 +08:00
parent c692339020
commit ce6dcf3600
2 changed files with 5 additions and 0 deletions

View File

@ -558,6 +558,8 @@ _register_template(
format_user=StringFormatter(slots=["<用户>{{content}}<AI>"]),
format_prefix=EmptyFormatter(slots=[{"bos_token"}]),
)
_register_template(
name="cpm3",
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
@ -566,6 +568,7 @@ _register_template(
stop_words=["<|im_end|>"],
)
_register_template(
name="dbrx",
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),

View File

@ -806,6 +806,7 @@ register_model_group(
template="cpm",
)
register_model_group(
models={
"MiniCPM3-4B": {
@ -816,6 +817,7 @@ register_model_group(
template="cpm3",
)
register_model_group(
models={
"Mistral-7B-v0.1": {