o òÜÓhÁã@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)Ú TensorTypec"súeZdZdZddgZdZdZ‡fdd„Z       d$d ee e e e e e fd e dee e efdee e efdeededeedee dee de de de de de deee efdef dd„Zdd„Zd d!„Zed"d#„ƒ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Ú tokenizerÚBridgeTowerImageProcessor)ÚRobertaTokenizerÚRobertaTokenizerFastcstƒ ||¡dS©N)ÚsuperÚ__init__)Úselfrr©Ú __class__©úl/var/www/html/ai/venv/lib/python3.10/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|jd|||||||| | | | | |||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. )rrrrrr r!r"r#r$r%r&r'r(r)T)r)Ú do_normalizeÚdo_center_cropNr)rrÚupdate)rÚimagesrrrrrr r!r"r#r$r%r&r'r(r)ÚkwargsÚencodingÚencoding_image_processorrrrÚ__call__1s8ñðÿÿÿ zBridgeTowerProcessor.__call__cOó|jj|i|¤Ž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/rrrr4eóz!BridgeTowerProcessor.batch_decodecOr3)z¾ This method forwards all its arguments to RobertaTokenizerFast's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )rÚdecoder5rrrr8lr7zBridgeTowerProcessor.decodecCs"|jj}|jj}tt ||¡ƒSr)rÚmodel_input_namesrÚlistÚdictÚfromkeys)rÚtokenizer_input_namesÚimage_processor_input_namesrrrr9ssz&BridgeTowerProcessor.model_input_names)NTFNNrNNNFFFFTN)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú attributesÚimage_processor_classÚtokenizer_classrrr r rÚboolÚstrrr rÚintr rr2r4r8Úpropertyr9Ú __classcell__rrrrr st ïýü û úùø ÷ ö õ ô óòñðï í4r N)rBÚtypingrrrÚprocessing_utilsrÚtokenization_utils_baserrr r r Úutilsr r rrrrÚs