sgKfdZddlmZmZmZmZddlmZmZm Z ddl m Z ddl m Z mZGdde Zy ) z Processor class for Nougat. )DictListOptionalUnion)PreTokenizedInput TextInputTruncationStrategy)ProcessorMixin)PaddingStrategy TensorTypecAeZdZdZddgZdZdZfdZ d-deded e e e fd d d ed ededede e e fdedee e ee fdee e ee fdeddee e dfdee eeeeeefde eeeeeefdee eeeeeefdede ee efde ee efdee de ded ee d!ee e efd"eed#eed$ed%ed&ed'ed(ef@d)Zd*Zd+Zd,ZxZS).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 tokenizerAutoImageProcessor AutoTokenizercHt||||j|_y)N)super__init__rcurrent_processor)selfrr __class__s _/var/www/html/venv/lib/python3.12/site-packages/transformers/models/nougat/processing_nougat.pyrzNougatProcessor.__init__-s  )4!%!5!5do_crop_margin do_resizesizeresamplePILImageResampling do_thumbnaildo_align_long_axisdo_pad do_rescalerescale_factor do_normalize image_mean image_std data_formatChannelDimensioninput_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#| | td|!|j|||||||| | | | | ||||}#|$|j||||||||||||||||| |!|"}$|#S|$S$d#d<|#S)NzBYou need to specify either an `images` or `text` input to process.)rrrrr!r"r#r$r%r&r'r(r6r)r+)r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r= input_idslabels) ValueErrorrr)%rimagestextrrrrr!r"r#r$r%r&r'r(r)r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=inputs encodingss% r__call__zNougatProcessor.__call__1sN >dlab b  ))-#!)#5%-)%#-'"3!*F$  #'!1#5%%$7#5-&;&;*C+E'=+''I, <M ^ (5F8 Mrc:|jj|i|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_decoderargskwargss rrHzNougatProcessor.batch_decodes +t~~**D;F;;rc:|jj|i|S)z This method forwards all its arguments to NougatTokenizer's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )rdecoderIs rrMzNougatProcessor.decodes %t~~$$d5f55rc:|jj|i|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_generationrIs rrOz'NougatProcessor.post_process_generations 6t~~55tFvFFr)"NNNNNNNNNNNNNNchannels_firstNNNNTFNNrFNNNNFFFFT)__name__ __module__ __qualname____doc__ attributesimage_processor_classtokenizer_classrboolrstrintrfloatrrrrr r r rFrHrMrO __classcell__)rs@rrrs $[1J0%O6  ##)-!#',0!:>9=4DFJmqei #'5:;?$($),0;?0404*/+0',#KZ Z  Z 38n Z'ZZ!ZZZc5j)ZZU5$u+#567ZE%e"456Z 01!Z"$E#/A*A$BC#Z$E)->YQUVgQh"hij%Z&9&7i$O`Jaab'Z(# ).YFWAXX Y )Z.!/Z0tS/121Z2$%7783Z4SM5Z67Z8"9Z:%SM;Z<!sJ!78=Z> (~?Z@ (~AZB$(CZD%)EZF!%GZHIZJKZx<6GrrN)rTtypingrrrr$transformers.tokenization_utils_baserrr processing_utilsr utilsr r rrrrbs./.aa.0EGnEGr