a  d@s`dZddlmZmZmZddlmZddlmZm Z m Z m Z m Z ddl mZGdddeZd S) z Processor class for Blip. )ListOptionalUnion)ProcessorMixin) BatchEncodingPaddingStrategyPreTokenizedInput TextInputTruncationStrategy) TensorTypecseZdZdZddgZdZdZfddZdee e e e e e fe ee e efee e efeeeeeee e e e e e e eee efed d dZddZddZeddZZS) 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 tokenizerZBlipImageProcessor)Z BertTokenizerZBertTokenizerFastcs d|_t|||j|_dS)NF)return_token_type_idssuper__init__rcurrent_processor)selfrr __class__q/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/transformers/models/blip/processing_blip.pyr+szBlipProcessor.__init__NTFr)textadd_special_tokenspadding truncation max_lengthstridepad_to_multiple_ofreturn_attention_maskreturn_overflowing_tokensreturn_special_tokens_maskreturn_offsets_mappingr return_lengthverbosereturn_tensorsreturncKs|dur|durtd|dur\|j|_|jf|||||||| | | | | |||d|}|S|j||d}|dur|jf|||||||| | | | | |||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.)rrrrrrrr r!r"r#rr$r%r&)r&) ValueErrorrrrupdate)rZimagesrrrrrrrr r!r"r#rr$r%r&kwargsZ text_encodingZencoding_image_processorrrr__call__0sb zBlipProcessor.__call__cOs|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*rrrr,szBlipProcessor.batch_decodecOs|jj|i|S)z This method forwards all its arguments to BertTokenizerFast's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )rdecoder-rrrr/szBlipProcessor.decodecCs"|jj}|jj}tt||S)N)rmodel_input_namesrlistdictfromkeys)rZtokenizer_input_namesZimage_processor_input_namesrrrr0szBlipProcessor.model_input_names)NNTFNNrNNFFFFFTN)__name__ __module__ __qualname____doc__ attributesZimage_processor_classZtokenizer_classrrr r rboolstrrr rintr rr+r,r/propertyr0 __classcell__rrrrr sV     Rr N)r7typingrrrZprocessing_utilsrZtokenization_utils_baserrr r r utilsr r rrrrs