a  þdàã@sPdZddlZddlmZddlmZddlmZddlm Z Gd d „d eƒZ dS) z% Speech processor class for Wav2Vec2 éN)Úcontextmanageré)ÚProcessorMixiné)ÚWav2Vec2FeatureExtractor)ÚWav2Vec2CTCTokenizercsdeZdZdZdZdZ‡fdd„Ze‡fdd„ƒZdd „Z d d „Z d d „Z dd„Z e dd„ƒZ‡ZS)ÚWav2Vec2ProcessoraŸ Constructs a Wav2Vec2 processor which wraps a Wav2Vec2 feature extractor and a Wav2Vec2 CTC tokenizer into a single processor. [`Wav2Vec2Processor`] offers all the functionalities of [`Wav2Vec2FeatureExtractor`] and [`PreTrainedTokenizer`]. See the docstring of [`~Wav2Vec2Processor.__call__`] and [`~Wav2Vec2Processor.decode`] for more information. Args: feature_extractor (`Wav2Vec2FeatureExtractor`): An instance of [`Wav2Vec2FeatureExtractor`]. The feature extractor is a required input. tokenizer ([`PreTrainedTokenizer`]): An instance of [`PreTrainedTokenizer`]. The tokenizer is a required input. rZ AutoTokenizercs tƒ ||¡|j|_d|_dS)NF)ÚsuperÚ__init__Úfeature_extractorÚcurrent_processorÚ_in_target_context_manager)Úselfr Ú tokenizer©Ú __class__©úy/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/transformers/models/wav2vec2/processing_wav2vec2.pyr +szWav2Vec2Processor.__init__c stztƒj|fi|¤ŽWStynt d|j›dt¡tj|fi|¤Ž}tj|fi|¤Ž}|||dYS0dS)NzLoading a tokenizer inside a from a config that does not include a `tokenizer_class` attribute is deprecated and will be removed in v5. Please add `'tokenizer_class': 'Wav2Vec2CTCTokenizer'` attribute to either your `config.json` or `tokenizer_config.json` file to suppress this warning: )r r) r Úfrom_pretrainedÚOSErrorÚwarningsÚwarnÚ__name__Ú FutureWarningrr)ÚclsZpretrained_model_name_or_pathÚkwargsr rrrrr0s  ú z!Wav2Vec2Processor.from_pretrainedcOsú|jr|j|i|¤ŽSd|vr4t d¡| d¡}n | dd¡}| dd¡}| dd¡}t|ƒdkrx|d}|dd…}|dur|durtd ƒ‚|dur´|j|g|¢Rd|i|¤Ž}|durÎ|j|fi|¤Ž}|durÚ|S|duræ|S|d |d <|SdS) a¹ When used in normal mode, this method forwards all its arguments to Wav2Vec2FeatureExtractor's [`~Wav2Vec2FeatureExtractor.__call__`] and returns its output. If used in the context [`~Wav2Vec2Processor.as_target_processor`] this method forwards all its arguments to PreTrainedTokenizer's [`~PreTrainedTokenizer.__call__`]. Please refer to the docstring of the above two methods for more information. Z raw_speechzLUsing `raw_speech` as a keyword argument is deprecated. Use `audio` instead.ÚaudioNÚ sampling_rateÚtextrrzAYou need to specify either an `audio` or `text` input to process.Ú input_idsÚlabels) r r rrÚpopÚlenÚ ValueErrorr r)rÚargsrrrrÚinputsÚ encodingsrrrÚ__call__Cs.        zWav2Vec2Processor.__call__cOsº|jr|jj|i|¤ŽS| dd¡}| dd¡}t|ƒdkrP|d}|dd…}|durr|jj|g|¢Ri|¤Ž}|durŽ|jj|fi|¤Ž}|durš|S|dur¦|S|d|d<|SdS)a¯ When used in normal mode, this method forwards all its arguments to Wav2Vec2FeatureExtractor's [`~Wav2Vec2FeatureExtractor.pad`] and returns its output. If used in the context [`~Wav2Vec2Processor.as_target_processor`] this method forwards all its arguments to PreTrainedTokenizer's [`~PreTrainedTokenizer.pad`]. Please refer to the docstring of the above two methods for more information. Úinput_featuresNr rrr)r r Úpadr!r"r r)rr$rr(r rrrr)is"     zWav2Vec2Processor.padcOs|jj|i|¤ŽS)zà This method forwards all its arguments to PreTrainedTokenizer's [`~PreTrainedTokenizer.batch_decode`]. Please refer to the docstring of this method for more information. )rÚ batch_decode©rr$rrrrr*‡szWav2Vec2Processor.batch_decodecOs|jj|i|¤ŽS)z½ This method forwards all its arguments to PreTrainedTokenizer's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )rÚdecoder+rrrr,ŽszWav2Vec2Processor.decodeccs0t d¡d|_|j|_dV|j|_d|_dS)zŒ Temporarily sets the tokenizer for processing the input. Useful for encoding the labels when fine-tuning Wav2Vec2. zî`as_target_processor` is deprecated and will be removed in v5 of Transformers. You can process your labels by using the argument `text` of the regular `__call__` method (either in the same call as your audio inputs, or in a separate call.TNF)rrr rr r )rrrrÚas_target_processor•sÿz%Wav2Vec2Processor.as_target_processor)rÚ __module__Ú __qualname__Ú__doc__Zfeature_extractor_classZtokenizer_classr Ú classmethodrr'r)r*r,rr-Ú __classcell__rrrrrs  &r) r0rÚ contextlibrZprocessing_utilsrZfeature_extraction_wav2vec2rZtokenization_wav2vec2rrrrrrÚs