U ,-e @s$dZddlmZGdddeZdS)z Processor class for TVLT. )ProcessorMixincsFeZdZdZddgZdZdZfddZdd d Ze d d Z Z S) TvltProcessora` Constructs a TVLT processor which wraps a TVLT image processor and TVLT feature extractor into a single processor. [`TvltProcessor`] offers all the functionalities of [`TvltImageProcessor`] and [`TvltFeatureExtractor`]. See the docstring of [`~TvltProcessor.__call__`] for more information. Args: image_processor (`TvltImageProcessor`): An instance of [`TvltImageProcessor`]. The image processor is a required input. feature_extractor (`TvltFeatureExtractor`): An instance of [`TvltFeatureExtractor`]. The feature extractor is a required input. image_processorfeature_extractorZTvltImageProcessorZTvltFeatureExtractorcs tj||d||_||_dS)N)rr)super__init__rr)selfrr __class__i/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/transformers/models/tvlt/processing_tvlt.pyr'szTvltProcessor.__init__NFc Os|dkr|dkrtdd} |dk r>|j|f|d|i|} |dk r`|j|f|ddi|} |dk r|j|f|||d|} i} |dk r| | |dk r| | | dk r| | | S)a Forwards the `images` argument to TvltImageProcessor's [`~TvltImageProcessor.preprocess`] and the `audio` argument to TvltFeatureExtractor's [`~TvltFeatureExtractor.__call__`]. Please refer to the docstring of the above two methods for more information. NzCYou need to specify either an `images` or `audio` input to process. mask_pixelZis_mixedT) sampling_rate mask_audio) ValueErrorrrupdate) rZimagesZaudioZ images_mixedrrr argskwargsZimages_mixed_dictZ images_dictZ audio_dictZ output_dictr r r __call__-s4   zTvltProcessor.__call__cCs"|jj}|jj}tt||S)N)rmodel_input_namesrlistdictfromkeys)rZimage_processor_input_namesZfeature_extractor_input_namesr r r rTszTvltProcessor.model_input_names)NNNNFF) __name__ __module__ __qualname____doc__ attributesZimage_processor_classZfeature_extractor_classrrpropertyr __classcell__r r r r rs   'rN)rZprocessing_utilsrrr r r r s