sg0pdZddlmZmZddlmZmZmZddlm Z m Z m Z Gdded Z Gd d eZ y ) z" Processor class for BridgeTower. )ListUnion)ProcessingKwargsProcessorMixinUnpack) BatchEncodingPreTokenizedInput TextInputc .eZdZdddddddddddddZy)BridgeTowerProcessorKwargsTFr)add_special_tokenspaddingstridereturn_overflowing_tokensreturn_special_tokens_maskreturn_offsets_mapping return_lengthverbose) do_normalizedo_center_crop) text_kwargs images_kwargsN)__name__ __module__ __qualname__ _defaultsi/var/www/html/venv/lib/python3.12/site-packages/transformers/models/bridgetower/processing_bridgetower.pyr r s5#').*/&+"  !" Irr F)totalceZdZdZddgZdZdZfdZ ddee e e e e e fde e d efd Zd Zd Zed ZxZS)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)RobertaTokenizerRobertaTokenizerFastc&t|||yN)super__init__)selfr$r% __class__s r r,zBridgeTowerProcessor.__init__@s )4rtextkwargsreturnc |jtfd|jji|}|jdd|i|d}|j|fi|d}|j ||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/rrr) _merge_kwargsr r% init_kwargsr$update) r-imagesr/audiovideosr0 output_kwargsencodingencoding_image_processors r __call__zBridgeTowerProcessor.__call__Cs+** & "&.."<"<  "4>>LtL}]/KL#74#7#7#a-P_B`#a 01rc:|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-argsr0s r r?z!BridgeTowerProcessor.batch_decode]s +t~~**D;F;;rc:|jj|i|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@s r rCzBridgeTowerProcessor.decodeds %t~~$$d5f55rc|jj}|jj}ttj ||zSr*)r%model_input_namesr$listdictfromkeys)r-tokenizer_input_namesimage_processor_input_namess r rEz&BridgeTowerProcessor.model_input_namesks? $ @ @&*&:&:&L&L#DMM"7:U"UVWWr)NNN)rrr__doc__ attributesimage_processor_classtokenizer_classr,rr r rrr r r=r?rCpropertyrE __classcell__)r.s@r r#r#,s $[1J7BO5 _c I0$y/4HYCZZ[ 34  4<6XXrr#N)rKtypingrrprocessing_utilsrrrtokenization_utils_baser r r r r#rrr rTs:HHRR!1&CX>CXr