o
    wi                     @   s,   d dl mZmZ dZdZG dd deZdS )    )ModalityPromptFormatterz<|im_start|>z
<|im_end|>c                   @   sZ   e Zd ZdZdZe dZde de ddej	idee d	e ddej	idiZ
d
S )QwenPromptFormatterqwen	assistantz
assistant
userzuser
|message|
message)templateslotsz	|message|N)__name__
__module____qualname__NAMEOUTPUT_ROLEQWEN_BOTINFERENCE_PREFIXQWEN_EOTr   TextTEMPLATE r   r   a/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/nemo/collections/common/prompts/qwen.pyr      s    
r   N))nemo.collections.common.prompts.formatterr   r   r   r   r   r   r   r   r   <module>   s   