o
    Zh                     @   s   d Z ddlmZmZ ddlmZmZ ddlmZ ddl	m
Z
 ddlmZ e r-ddlmZ G dd	 d	eZe
d
dG dd deZdgZdS )z(Fast Video processor class for InternVL.   )OPENAI_CLIP_MEANOPENAI_CLIP_STD)UnpackVideosKwargs)is_vision_available)requires)BaseVideoProcessor)PILImageResamplingc                   @   s   e Zd ZdS ) InternVLVideoProcessorInitKwargsN)__name__
__module____qualname__ r   r   e/var/www/auris/lib/python3.10/site-packages/transformers/models/internvl/video_processing_internvl.pyr
   #   s    r
   )Ztorchvision)backendsc                       sX   e Zd ZejZeZeZ	dddZ
dZdZdZdZeZdgZdee f fddZ  ZS )InternVLVideoProcessori  )heightwidthTZpixel_values_videoskwargsc                    s   t  jdi | d S )Nr   )super__init__)selfr   	__class__r   r   r   3   s   zInternVLVideoProcessor.__init__)r   r   r   r	   ZBICUBICZresampler   Z
image_meanr   Z	image_stdsizeZ	do_resizeZ
do_rescaleZdo_normalizeZdo_convert_rgbr
   Zvalid_kwargsZmodel_input_namesr   r   __classcell__r   r   r   r   r   &   s    
r   N)__doc__Zimage_utilsr   r   Zprocessing_utilsr   r   utilsr   Zutils.import_utilsr   Zvideo_processing_utilsr   r	   r
   r   __all__r   r   r   r   <module>   s   
