tjxdZddlmZmZddlmZGddedZeGdd eZd gZd S) z Processor class for TVP. )ProcessingKwargsProcessorMixin)auto_docstringc eZdZddddddiZdS)TvpProcessorKwargs text_kwargsT max_lengthF) truncationpaddingpad_to_max_lengthreturn_token_type_idsN)__name__ __module__ __qualname__ _defaultsd/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/transformers/models/tvp/processing_tvp.pyrrs.#!%%*   IIIrrF)totalc&eZdZdfd ZdZxZS) TvpProcessorNc Zt||||_dS)N)super__init__video_processor)selfimage_processor tokenizerkwargs __class__s rrzTvpProcessor.__init__#s+ )444.rct|dd|zdt|dd|zd}}||fS)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. )roundtolist)rlogitsvideo_durationsstartends rpost_process_video_groundingz)TvpProcessor.post_process_video_grounding's_" &--//!$Q'/91 = = &--//!$Q'/91 = = czr)NN)rrrrr* __classcell__)r s@rrr!sL//////rrN) __doc__processing_utilsrrutilsrrr__all__rrrr0sA@@@@@@@######)>:  r