o wiO@sfdZddlmZddlmZmZmZddlmZm Z m Z Gdddedd Z Gd d d eZ d gZ d S) z" Processor class for BridgeTower. )Union)ProcessingKwargsProcessorMixinUnpack) BatchEncodingPreTokenizedInput TextInputc @s.eZdZdddddddddddddZdS)BridgeTowerProcessorKwargsTFr)add_special_tokenspaddingstridereturn_overflowing_tokensreturn_special_tokens_maskreturn_offsets_mapping return_lengthverbose) do_normalizedo_center_crop) text_kwargs images_kwargsN)__name__ __module__ __qualname__ _defaultsrrs/home/ubuntu/sommelier/.venv/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 tokenizerBridgeTowerImageProcessor)RobertaTokenizerRobertaTokenizerFastcst||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. tokenizer_init_kwargsr*rrNr) _merge_kwargsr r init_kwargsrupdate) r'imagesr*audiovideosr+ output_kwargsencodingencoding_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_decoder'argsr+rrrr9]z!BridgeTowerProcessor.batch_decodecOr8)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=dr<zBridgeTowerProcessor.decodecCs"|jj}|jj}tt||Sr$)r model_input_namesrlistdictfromkeys)r'tokenizer_input_namesimage_processor_input_namesrrrr>ksz&BridgeTowerProcessor.model_input_names)NNN)rrr__doc__ attributesimage_processor_classtokenizer_classr&rr rr?rr rr7r9r=propertyr> __classcell__rrr(rr,s(  rN)rDtypingrprocessing_utilsrrrtokenization_utils_baserrr r r__all__rrrrs  F