Skip to content

Commit c882289

Browse files
author
xusenlin
committed
Update huatuo template
1 parent 96229b5 commit c882289

File tree

1 file changed

+33
-0
lines changed

1 file changed

+33
-0
lines changed

api/apapter/template.py

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -896,6 +896,38 @@ def template(self):
896896
)
897897

898898

899+
class HuatuoTemplate(BaseTemplate):
900+
901+
name = "huatuo"
902+
allow_models = ["huatuo"]
903+
system_prompt = "一位用户和智能医疗大模型HuatuoGPT之间的对话。对于用户的医疗问诊,HuatuoGPT给出准确的、详细的、温暖的指导建议。对于用户的指令问题,HuatuoGPT给出有益的、详细的、有礼貌的回答。"
904+
stop = {
905+
"strings": ["<reserved_102>", "<reserved_103>", "<病人>"],
906+
"token_ids": [195, 196],
907+
}
908+
909+
@property
910+
def template(self):
911+
return (
912+
"{% set found_item = false %}"
913+
"{% for message in messages %}"
914+
"{% if message['role'] == 'system' %}"
915+
"{% set found_item = true %}"
916+
"{% endif %}"
917+
"{% endfor %}"
918+
"{% if not found_item %}"
919+
"{{ system_prompt }}"
920+
"{% endif %}"
921+
"{% for message in messages %}"
922+
"{% if message['role'] == 'user' %}"
923+
"{{ '<病人>:' + message['content'] + ' <HuatuoGPT>:' }}"
924+
"{% elif message['role'] == 'assistant' %}"
925+
"{{ message['content'] + '</s>' }}"
926+
"{% endif %}"
927+
"{% endfor %}"
928+
)
929+
930+
899931
register_prompt_adapter(AlpacaTemplate)
900932
register_prompt_adapter(AquilaChatTemplate)
901933
register_prompt_adapter(BaiChuanTemplate)
@@ -908,6 +940,7 @@ def template(self):
908940
register_prompt_adapter(DeepseekTemplate)
909941
register_prompt_adapter(FireflyTemplate)
910942
register_prompt_adapter(FireflyForQwenTemplate)
943+
register_prompt_adapter(HuatuoTemplate)
911944
register_prompt_adapter(InternLMTemplate)
912945
register_prompt_adapter(Llama2Template)
913946
register_prompt_adapter(MossTemplate)

0 commit comments

Comments
 (0)