o wi@s^ddlmZmZddlmZmZeGdddZeGdddeZeGdddeZd S) ) dataclassfield)ListOptionalc@sPeZdZUdZdZeeed<edddZ e eed<dZ eeed <dZ dS) BaseConversationTemplateConfigz/Conversation template config related parameterssystemcCddgSNuser assistantr r r r/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/nemo/collections/multimodal/data/energon/conversation.pyz'BaseConversationTemplateConfig.default_factoryrolesN stop_string) __name__ __module__ __qualname____doc__rrstr__annotations__rrrr chat_templater r r rrs rc@sbeZdZUdZeddZeed<edddZe eed<ed dZ eed <ed dZ eed <d S)LLaVATemplateConfigzCLLava-specific template configuration which extends the base configzA chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed and polite answers to user's questions.defaultrcCr r r r r r rr$rzLLaVATemplateConfig.rrzra {%- for message in messages %} {%- if message['role'] == 'system' %} {{- message['content'].strip() + ' ' -}} {%- elif message['role'] == 'user' %} {{- 'USER: ' -}} {{- message['content'].strip() + ' ' -}} {%- elif message['role'] == 'assistant' %} {{- 'ASSISTANT: ' -}} {{- message['content'].strip() -}} {{- '' -}} {%- endif %} {%- endfor -%} rN) rrrrrrrrrrrrr r r rrs rc@sPeZdZUdZeddZeed<edddZe eed<ed dZ eed <dS) MLlamaTemplateConfigzDMLlama specific template configuration which extends the base configNrrcCr r r r r r rr;rzMLlamaTemplateConfig.rrz <|eot_id|>r) rrrrrrrrrrrr r r rr6s rN) dataclassesrrtypingrrrrrr r r rs