o h=@sldZddlmZmZmZddlmZddlmZddl m Z m Z m Z m Z mZddlmZGdd d eZd S) z Processor class for Blip. )ListOptionalUnion) ImageInput)ProcessorMixin) BatchEncodingPaddingStrategyPreTokenizedInput TextInputTruncationStrategy) TensorTypec$seZdZdZddgZdZdZfddZ       d%d ed e e e e e e e fde de e eefde e eefdeededeedee de de de de de de dee eefdef"ddZdd Zd!d"Zed#d$ZZS)& BlipProcessora] Constructs a BLIP processor which wraps a BERT tokenizer and BLIP image processor into a single processor. [`BlipProcessor`] offers all the functionalities of [`BlipImageProcessor`] and [`BertTokenizerFast`]. See the docstring of [`~BlipProcessor.__call__`] and [`~BlipProcessor.decode`] for more information. Args: image_processor (`BlipImageProcessor`): An instance of [`BlipImageProcessor`]. The image processor is a required input. tokenizer (`BertTokenizerFast`): An instance of ['BertTokenizerFast`]. The tokenizer is a required input. image_processor tokenizerBlipImageProcessor) BertTokenizerBertTokenizerFastcs d|_t|||j|_dS)NF)return_token_type_idssuper__init__rcurrent_processor)selfrr __class__^/var/www/html/ai/venv/lib/python3.10/site-packages/transformers/models/blip/processing_blip.pyr-s zBlipProcessor.__init__NTFrimagestextadd_special_tokenspadding truncation max_lengthstridepad_to_multiple_ofreturn_attention_maskreturn_overflowing_tokensreturn_special_tokens_maskreturn_offsets_mappingr return_lengthverbosereturn_tensorsreturncKs|dur |dur td|dur.|j|_|jd|||||||| | | | | |||d|}|S|j||d}|durR|jd|||||||| | | | | |||d|}nd}|dur]|||S)a  This method uses [`BlipImageProcessor.__call__`] method to prepare image(s) for the model, and [`BertTokenizerFast.__call__`] to prepare text for the model. Please refer to the docstring of the above two methods for more information. Nz*You have to specify either images or text.)rrr r!r"r#r$r%r&r'r(rr)r*r+)r+r) ValueErrorrrrupdate)rrrrr r!r"r#r$r%r&r'r(rr)r*r+kwargs text_encodingencoding_image_processorrrr__call__2sb zBlipProcessor.__call__cO|jj|i|S)z This method forwards all its arguments to BertTokenizerFast's [`~PreTrainedTokenizer.batch_decode`]. Please refer to the docstring of this method for more information. )r batch_decoderargsr/rrrr4zBlipProcessor.batch_decodecOr3)z This method forwards all its arguments to BertTokenizerFast's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )rdecoder5rrrr8r7zBlipProcessor.decodecCs"|jj}|jj}tt||S)N)rmodel_input_namesrlistdictfromkeys)rtokenizer_input_namesimage_processor_input_namesrrrr9szBlipProcessor.model_input_names)NNTFNNrNNFFFFFTN)__name__ __module__ __qualname____doc__ attributesimage_processor_classtokenizer_classrrrr r rboolstrr r rintr rr2r4r8propertyr9 __classcell__rrrrrsz          RrN)rBtypingrrr image_utilsrprocessing_utilsrtokenization_utils_baserr r r r utilsr rrrrrs