g0~dZddlmZmZddlmZmZmZddlm Z m Z m Z Gdded Z Gd d eZ d S) z" Processor class for BridgeTower. )ListUnion)ProcessingKwargsProcessorMixinUnpack) BatchEncodingPreTokenizedInput TextInputc 0eZdZdddddddddddddZdS)BridgeTowerProcessorKwargsTFr)add_special_tokenspaddingstridereturn_overflowing_tokensreturn_special_tokens_maskreturn_offsets_mapping return_lengthverbose) do_normalizedo_center_crop) text_kwargs images_kwargsN)__name__ __module__ __qualname__ _defaultsr/var/www/html/ai-engine/env/lib/python3.11/site-packages/transformers/models/bridgetower/processing_bridgetower.pyr r sP#').*/&+"   !"  IIIrr F)totalceZdZdZddgZdZdZfdZ ddee e e e e e fd e e d efd Zd Zd ZedZxZS)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)RobertaTokenizerRobertaTokenizerFastcLt||dSN)super__init__)selfr$r% __class__s r r,zBridgeTowerProcessor.__init__@s# )44444rNtextkwargsreturnc |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/rrr) _merge_kwargsr r% init_kwargsr$update) r-imagesr/audiovideosr0 output_kwargsencodingencoding_image_processors r __call__zBridgeTowerProcessor.__call__Cs+* &  "&."<   "4>LLtL}]/KLL#74#7#a#a-P_B`#a#a 0111rc&|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~$d5f555rc|jj}|jj}tt||zSr*)r%model_input_namesr$listdictfromkeys)r-tokenizer_input_namesimage_processor_input_namess r rEz&BridgeTowerProcessor.model_input_namesks: $ @&*&:&L#DMM"7:U"UVVWWWr)NNN)rrr__doc__ attributesimage_processor_classtokenizer_classr,rr r rrr r r=r?rCpropertyrE __classcell__)r.s@r r#r#,s  $[1J7BO55555 _c I0$y/4HYCZZ[ 34  4<<<666XXXXXXXXrr#N)rKtypingrrprocessing_utilsrrrtokenization_utils_baser r r r r#rrr rTsHHHHHHHHHHRRRRRRRRRR!1&CXCXCXCXCX>CXCXCXCXCXr