Ë
    ÜÓ^jµ  ã                   ód   — d Z ddlmZmZ ddlmZ  G d„ ded¬«      Ze G d„ d	e«      «       Zd	gZy
)z
Processor class for TVP.
é   )ÚProcessingKwargsÚProcessorMixin)Úauto_docstringc                   ó   — e Zd ZddddddœiZy)ÚTvpProcessorKwargsÚtext_kwargsTÚ
max_lengthF)Ú
truncationÚpaddingÚpad_to_max_lengthÚreturn_token_type_idsN)Ú__name__Ú
__module__Ú__qualname__Ú	_defaults© ó    úq/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/transformers/models/tvp/processing_tvp.pyr   r      s   „ àØØ#Ø!%Ø%*ñ	
ð�Ir   r   F)Útotalc                   ó&   ‡ — e Zd Zdˆ fd„	Zd„ Zˆ xZS )ÚTvpProcessorc                 ó4   •— t         ‰| �  ||«       || _        y )N)ÚsuperÚ__init__Úvideo_processor)ÚselfÚimage_processorÚ	tokenizerÚkwargsÚ	__class__s       €r   r   zTvpProcessor.__init__#   s   ø€ Ü‰Ñ˜¨)Ô4Ø.ˆÕr   c                 ó–   — t        |j                  «       d   d   |z  d«      t        |j                  «       d   d   |z  d«      }}||fS )aŽ  
        Compute the time of the video.

        Args:
            logits (`torch.Tensor`):
                The logits output of TvpForVideoGrounding.
            video_durations (`float`):
                The video's duration.

        Returns:
            start (`float`):
                The start time of the video.
            end (`float`):
                The end time of the video.
        é    é   )ÚroundÚtolist)r   ÚlogitsÚvideo_durationsÚstartÚends        r   Úpost_process_video_groundingz)TvpProcessor.post_process_video_grounding'   sU   € ô" �&—-‘-“/ !Ñ$ QÑ'¨/Ñ9¸1Ó=Ü�&—-‘-“/ !Ñ$ QÑ'¨/Ñ9¸1Ó=ð ˆð
 �cˆzÐr   )NN)r   r   r   r   r*   Ú__classcell__)r    s   @r   r   r   !   s   ø„ õ/ör   r   N)	Ú__doc__Úprocessing_utilsr   r   Úutilsr   r   r   Ú__all__r   r   r   ú<module>r0      sG   ðñ÷ AÝ #ôÐ)°õ ð ô�>ó ó ðð: Ð
�r   