U 9%eR@s`dZddlmZmZmZmZddlmZmZm Z ddl m Z ddl m Z mZGddde Zd S) z Processor class for Nougat. )DictListOptionalUnion)PreTokenizedInput TextInputTruncationStrategy)ProcessorMixin)PaddingStrategy TensorTypec"sJeZdZdZddgZdZdZfddZdeee e e fd eeeee e e feee e ee fee e ee fedee e dfee eeeeeefe eeeeeefee eeeeeefee ee efe ee efee e eee ee e efeeeeeeeeed ddZddZddZddZZS)NougatProcessoran Constructs a Nougat processor which wraps a Nougat image processor and a Nougat tokenizer into a single processor. [`NougatProcessor`] offers all the functionalities of [`NougatImageProcessor`] and [`NougatTokenizerFast`]. See the [`~NougatProcessor.__call__`] and [`~NougatProcessor.decode`] for more information. Args: image_processor ([`NougatImageProcessor`]): An instance of [`NougatImageProcessor`]. The image processor is a required input. tokenizer ([`NougatTokenizerFast`]): An instance of [`NougatTokenizerFast`]. The tokenizer is a required input. image_processor tokenizerZAutoImageProcessorZ AutoTokenizercst|||j|_dS)N)super__init__rZcurrent_processor)selfrr __class__k/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/transformers/models/nougat/processing_nougat.pyr,szNougatProcessor.__init__NChannelDimension.FIRSTTFrZPILImageResamplingZChannelDimension) do_crop_margin do_resizesizeresample do_thumbnaildo_align_long_axisdo_pad do_rescalerescale_factor do_normalize image_mean image_std data_formatinput_data_format text_pair text_targettext_pair_targetadd_special_tokenspadding truncation max_lengthstrideis_split_into_wordspad_to_multiple_ofreturn_tensorsreturn_token_type_idsreturn_attention_maskreturn_overflowing_tokensreturn_special_tokens_maskreturn_offsets_mapping return_lengthverbosec#%Cs|dkr|dkrtd|dk rJ|j|||||||| | | | | ||||d}#|dk r|j||||||||||||||||| |!|"d}$|dkr|#S|dkr|$S|$d|#d<|#SdS)NzBYou need to specify either an `images` or `text` input to process.)rrrrrrrrr r!r"r#r0r$r%)r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7Z input_idslabels) ValueErrorrr)%rZimagestextrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7inputs encodingsrrr__call__0sb' zNougatProcessor.__call__cOs|jj||S)z This method forwards all its arguments to NougatTokenizer's [`~PreTrainedTokenizer.batch_decode`]. Please refer to the docstring of this method for more information. )r batch_decoderargskwargsrrrr>szNougatProcessor.batch_decodecOs|jj||S)z This method forwards all its arguments to NougatTokenizer's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )rdecoder?rrrrBszNougatProcessor.decodecOs|jj||S)z This method forwards all its arguments to NougatTokenizer's [`~PreTrainedTokenizer.post_process_generation`]. Please refer to the docstring of this method for more information. )rpost_process_generationr?rrrrCsz'NougatProcessor.post_process_generation)"NNNNNNNNNNNNNNrNNNNTFNNrFNNNNFFFFT)__name__ __module__ __qualname____doc__ attributesZimage_processor_classZtokenizer_classrboolrstrintrfloatrrrrr rr r=r>rBrC __classcell__rrrrr s       \r N)rGtypingrrrrZ$transformers.tokenization_utils_baserrrZprocessing_utilsr utilsr r r rrrrs