U ,È-eÀã@s`dZddlmZmZmZddlmZddlmZm Z m Z m Z m Z ddl mZGdd„deƒZd S) z" Processor class for BridgeTower. é)ÚListÚOptionalÚUnioné)ÚProcessorMixin)Ú BatchEncodingÚPaddingStrategyÚPreTokenizedInputÚ TextInputÚTruncationStrategy)Ú TensorTypecs¾eZdZdZddgZdZdZ‡fdd„Zdee e e e e e fe ee e efee e efeeeeeee ee e e e e e eee efed œd d„Zdd„Zdd„Zedd„ƒZ‡ZS)ÚBridgeTowerProcessora© Constructs a BridgeTower processor which wraps a Roberta tokenizer and BridgeTower image processor into a single processor. [`BridgeTowerProcessor`] offers all the functionalities of [`BridgeTowerImageProcessor`] and [`RobertaTokenizerFast`]. See the docstring of [`~BridgeTowerProcessor.__call__`] and [`~BridgeTowerProcessor.decode`] for more information. Args: image_processor (`BridgeTowerImageProcessor`): An instance of [`BridgeTowerImageProcessor`]. The image processor is a required input. tokenizer (`RobertaTokenizerFast`): An instance of ['RobertaTokenizerFast`]. The tokenizer is a required input. Úimage_processorÚ tokenizerZBridgeTowerImageProcessor)ZRobertaTokenizerZRobertaTokenizerFastcstƒ ||¡dS©N)ÚsuperÚ__init__)Úselfrr©Ú __class__©úw/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/transformers/models/bridgetower/processing_bridgetower.pyr-szBridgeTowerProcessor.__init__NTFr)ÚtextÚadd_special_tokensÚpaddingÚ truncationÚ max_lengthÚstrideÚpad_to_multiple_ofÚreturn_token_type_idsÚreturn_attention_maskÚreturn_overflowing_tokensÚreturn_special_tokens_maskÚreturn_offsets_mappingÚ return_lengthÚverboseÚreturn_tensorsÚreturncKsX|jf|||||||| | | | | |||dœ|—Ž}|j|f|dddœ|—Ž}| |¡|S)a This method uses [`BridgeTowerImageProcessor.__call__`] method to prepare image(s) for the model, and [`RobertaTokenizerFast.__call__`] to prepare text for the model. Please refer to the docstring of the above two methods for more information. )rrrrrrrrr r!r"r#r$r%r&T)r&Z do_normalizeZdo_center_crop)rrÚupdate)rZimagesrrrrrrrrr r!r"r#r$r%r&ÚkwargsÚencodingZencoding_image_processorrrrÚ__call__0s<ñðÿÿÿ zBridgeTowerProcessor.__call__cOs|jj||ŽS)zÄ This method forwards all its arguments to RobertaTokenizerFast's [`~PreTrainedTokenizer.batch_decode`]. Please refer to the docstring of this method for more information. )rÚ batch_decode©rÚargsr)rrrr,dsz!BridgeTowerProcessor.batch_decodecOs|jj||ŽS)z¾ This method forwards all its arguments to RobertaTokenizerFast's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )rÚdecoder-rrrr/kszBridgeTowerProcessor.decodecCs"|jj}|jj}tt ||¡ƒSr)rÚmodel_input_namesrÚlistÚdictÚfromkeys)rZtokenizer_input_namesZimage_processor_input_namesrrrr0rsz&BridgeTowerProcessor.model_input_names)NTFNNrNNNFFFFTN)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú attributesZimage_processor_classZtokenizer_classrrr r rÚboolÚstrrr rÚintr rr+r,r/Úpropertyr0Ú __classcell__rrrrr sT ï  í 4r N)r7ÚtypingrrrZprocessing_utilsrZtokenization_utils_baserrr r r Úutilsr r rrrrÚs