o
    }o™i0  ã                   @   sH   d Z ddlmZmZ G dd„ deƒZG dd„ deƒZG dd„ deƒZd	S )
z[
Implemented following the guide at https://www.promptingguide.ai/models/phi-2#phi-2-usage
é    )ÚModalityÚPromptFormatterc                   @   ó8   e Zd ZdZdZdddejidœeddejidœiZdS )	ÚPhi2QAPromptFormatterÚphi2_qaÚ	assistantÚuserzInstruct: |message|
Output: Úmessage©ÚtemplateÚslotsú	|message|N©Ú__name__Ú
__module__Ú__qualname__ÚNAMEÚOUTPUT_ROLEr   ÚTextÚTEMPLATE© r   r   úX/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/common/prompts/phi2.pyr      ó    ÿþÿþùr   c                   @   r   )	ÚPhi2ChatPromptFormatterÚ	phi2_chatr   r   zHuman: |message|
AI: r	   r
   r   Nr   r   r   r   r   r   )   r   r   c                   @   r   )	ÚPhi2CodePromptFormatterÚ	phi2_coder   r   z
|message|
r	   r
   r   Nr   r   r   r   r   r   <   r   r   N)Ú__doc__Ú)nemo.collections.common.prompts.formatterr   r   r   r   r   r   r   r   r   Ú<module>   s
   