Source code for lmflow.utils.conversation_template.chatglm

#!/usr/bin/env python
# Copyright 2024 Statistics and Machine Learning Research Group. All rights reserved.
from .base import ConversationTemplate, StringFormatter, TemplateComponent

[docs] CHATGLM3_TEMPLATE = ConversationTemplate( template_name="chatglm3", user_formatter=StringFormatter(template=[TemplateComponent(type="string", content="<|user|>\n {{content}}")]), assistant_formatter=StringFormatter( template=[TemplateComponent(type="string", content="<|assistant|>\n {{content}}")] ), system_formatter=StringFormatter(template=[TemplateComponent(type="string", content="<|system|>\n {{content}}")]), special_starter=TemplateComponent(type="string", content="[gMASK]sop"), )