o ZhU@sjdZddlmZmZddlmZmZmZddlm Z m Z m Z Gdddedd Z Gd d d eZ d gZd S) z" Processor class for BridgeTower. )ListUnion)ProcessingKwargsProcessorMixinUnpack) BatchEncodingPreTokenizedInput TextInputc @s.eZdZdddddddddddddZdS)BridgeTowerProcessorKwargsTFr)Zadd_special_tokenspaddingZstrideZreturn_overflowing_tokensZreturn_special_tokens_maskZreturn_offsets_mappingZ return_lengthverbose)Z do_normalizeZdo_center_crop) text_kwargs images_kwargsN)__name__ __module__ __qualname__ _defaultsrre/var/www/auris/lib/python3.10/site-packages/transformers/models/bridgetower/processing_bridgetower.pyr s  r F)totalcseZdZdZddgZdZdZfddZ   dd ee e e e e e fd e e d efd d ZddZddZeddZZS)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||dSN)super__init__)selfrr __class__rrr@szBridgeTowerProcessor.__init__Ntextkwargsreturnc KsV|jtfd|jji|}|jdd|i|d}|j|fi|d}|||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. Ztokenizer_init_kwargsr rrNr)Z _merge_kwargsr rZ init_kwargsrupdate) rZimagesr ZaudioZvideosr!Z output_kwargsencodingZencoding_image_processorrrr__call__Cs 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_decoderargsr!rrrr']z!BridgeTowerProcessor.batch_decodecOr&)z This method forwards all its arguments to RobertaTokenizerFast's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )rdecoder(rrrr+dr*zBridgeTowerProcessor.decodecCs"|jj}|jj}tt||Sr)rmodel_input_namesrlistdictfromkeys)rZtokenizer_input_namesZimage_processor_input_namesrrrr,ksz&BridgeTowerProcessor.model_input_names)NNN)rrr__doc__ attributesZimage_processor_classZtokenizer_classrrr r rrr rr%r'r+propertyr, __classcell__rrrrr,s(  rN)r0typingrrZprocessing_utilsrrrZtokenization_utils_baserr r r r__all__rrrrs F