o iX @sBdZddlmZmZGdddeddZGdddeZdgZd S) z Processor class for TVP. )ProcessingKwargsProcessorMixinc@seZdZddddddiZdS)TvpProcessorKwargs text_kwargsT max_lengthF) truncationpaddingpad_to_max_lengthreturn_token_type_idsN)__name__ __module__ __qualname__ _defaultsrrc/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/transformers/models/tvp/processing_tvp.pyrsrF)totalcs:eZdZdZddgZdZdZd fdd Zd d ZZ S) TvpProcessora Constructs an TVP processor which wraps a TVP image processor and a Bert tokenizer into a single processor. [`TvpProcessor`] offers all the functionalities of [`TvpImageProcessor`] and [`BertTokenizerFast`]. See the [`~TvpProcessor.__call__`] and [`~TvpProcessor.decode`] for more information. Args: image_processor ([`TvpImageProcessor`], *optional*): The image processor is a required input. tokenizer ([`BertTokenizerFast`], *optional*): The tokenizer is a required input. image_processor tokenizerTvpImageProcessor) BertTokenizerBertTokenizerFastNc st||||_dS)N)super__init__video_processor)selfrrkwargs __class__rrr3s 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_grounding7sz)TvpProcessor.post_process_video_grounding)NN) r r r __doc__ attributesimage_processor_classtokenizer_classrr' __classcell__rrrrr!s rN)r(processing_utilsrrrr__all__rrrrs  .