-iddZddlmZmZddlmZGddedZeGdd eZd gZy ) z Processor class for TVP. )ProcessingKwargsProcessorMixin)auto_docstringceZdZddddddiZy)TvpProcessorKwargs text_kwargsT max_lengthF) truncationpaddingpad_to_max_lengthreturn_token_type_idsN)__name__ __module__ __qualname__ _defaultsb/mnt/e/genesis-system/.venv/lib/python3.12/site-packages/transformers/models/tvp/processing_tvp.pyrrs#!%%*  IrrF)totalc&eZdZdfd ZdZxZS) TvpProcessorc 4t|||||_y)N)super__init__video_processor)selfimage_processor tokenizerkwargs __class__s rrzTvpProcessor.__init__#s )4.rct|jdd|zdt|jdd|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'sU" &--/!$Q'/91 = &--/!$Q'/91 = czr)NN)rrrrr* __classcell__)r s@rrr!s /rrN) __doc__processing_utilsrrutilsrrr__all__rrrr0sGA#)>:  r