a h @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)Zadd_special_tokenspaddingZstrideZreturn_overflowing_tokensZreturn_special_tokens_maskZreturn_offsets_mappingZ return_lengthverbose)Z do_normalizeZdo_center_crop) text_kwargs images_kwargsN)__name__ __module__ __qualname__ _defaultsrrr/var/www/html/assistant/venv/lib/python3.9/site-packages/transformers/models/bridgetower/processing_bridgetower.pyr s r F)totalcs\eZdZdZddgZdZdZfddZd ee e e e e e fe e ed d d ZZS) 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||dS)N)super__init__)selfrr __class__rrr@szBridgeTowerProcessor.__init__N)textkwargsreturnc KsV|jtfd|jji|}|jfd|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_kwargsrr r)Z _merge_kwargsr rZ init_kwargsrupdate) rZimagesrZaudioZvideosrZ output_kwargsencodingZencoding_image_processorrrr__call__Cs zBridgeTowerProcessor.__call__)NNN)rrr__doc__ attributesZimage_processor_classZtokenizer_classrrr rlistrr rr# __classcell__rrrrr,s rN)r$typingrZprocessing_utilsrrrZtokenization_utils_baserrr r r__all__rrrrs  2