o
    wi                     @   sn   d dl mZmZ d dlmZ ddlmZmZmZ eG dd dZ	eG dd de	Z
eG d	d
 d
e	ZdS )    )	dataclassfield)Optional   )
ImageTokenMultiModalToken
VideoTokenc                   @   sN   e Zd ZU eed< eed< dZeed< dZeed< dZ	eed< dZ
eed< d	S )

DataConfig
media_typemedia_tokenv1conv_templateFreset_position_idsreset_attention_maskeod_mask_lossN)__name__
__module____qualname__str__annotations__r   r   r   boolr   r    r   r   b/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/nemo/collections/vlm/neva/data/config.pyr	      s   
 r	   c                   @   sL   e Zd ZU dZeed< edd dZeed< dZ	e
e ed< d	Zeed
< dS )ImageDataConfigimager
   c                   C      t  S N)r   r   r   r   r   <lambda>"       zImageDataConfig.<lambda>default_factoryr   Nimage_folderpadimage_process_mode)r   r   r   r
   r   r   r   r   r   r!   r   r#   r   r   r   r   r      s
   
 r   c                   @   sh   e Zd ZU dZeed< edd dZeed< dZ	e
e ed< d	Zeed
< dZeed< dZe
e ed< dS )VideoDataConfigvideor
   c                   C   r   r   )r   r   r   r   r   r   *   r   zVideoDataConfig.<lambda>r   r   Nsplice_single_frame   
num_framesFsep_token_between_framesvideo_folder)r   r   r   r
   r   r   r   r   r   r&   r   r(   intr)   r   r*   r   r   r   r   r$   '   s   
 r$   N)dataclassesr   r   typingr   multimodal_tokensr   r   r   r	   r   r$   r   r   r   r   <module>   s   	