@@ -896,6 +896,38 @@ def template(self):
896
896
)
897
897
898
898
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
+
899
931
register_prompt_adapter (AlpacaTemplate )
900
932
register_prompt_adapter (AquilaChatTemplate )
901
933
register_prompt_adapter (BaiChuanTemplate )
@@ -908,6 +940,7 @@ def template(self):
908
940
register_prompt_adapter (DeepseekTemplate )
909
941
register_prompt_adapter (FireflyTemplate )
910
942
register_prompt_adapter (FireflyForQwenTemplate )
943
+ register_prompt_adapter (HuatuoTemplate )
911
944
register_prompt_adapter (InternLMTemplate )
912
945
register_prompt_adapter (Llama2Template )
913
946
register_prompt_adapter (MossTemplate )
0 commit comments