o ei@sRdZddlmZmZddlmZGdddeddZeGdd d eZd gZd S) z Processor class for TVP. )ProcessingKwargsProcessorMixin)auto_docstringc@seZdZddddddiZdS)TvpProcessorKwargs text_kwargsT max_lengthF) truncationpaddingpad_to_max_lengthreturn_token_type_idsN)__name__ __module__ __qualname__ _defaultsrrd/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/transformers/models/tvp/processing_tvp.pyrsrF)totalcs&eZdZdfdd ZddZZS) TvpProcessorNc st||||_dS)N)super__init__video_processor)selfimage_processor tokenizerkwargs __class__rrr#s zTvpProcessor.__init__cCs>t|dd|dt|dd|d}}||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_durationsstartendrrrpost_process_video_grounding'sz)TvpProcessor.post_process_video_grounding)NN)r r rrr% __classcell__rrrrr!srN) __doc__processing_utilsrrutilsrrr__all__rrrrs