o ei @sjdZddlmZmZmZddlmZmZmZddl m Z GdddeddZ e Gd d d eZ d gZ d S) z% Speech processor class for Wav2Vec2 )ProcessingKwargsProcessorMixinUnpack) AudioInputPreTokenizedInput TextInput)auto_docstringc@seZdZiZdS)Wav2Vec2ProcessorKwargsN)__name__ __module__ __qualname__ _defaultsrrn/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/transformers/models/wav2vec2/processing_wav2vec2.pyr sr F)totalc sleZdZfddZe  d dedBdeeeBeBe BdBde e fddZ d d Z ed d ZZS)Wav2Vec2Processorcst||dS)N)super__init__)selffeature_extractor tokenizer __class__rrrszWav2Vec2Processor.__init__NaudiotextkwargscKs|dur |dur td|jtfd|jji|}|dur(|j|fi|d}|dur7|j|fi|d}|dur=|S|durC|S|d|d<|S)z Returns: This method returns the results of each `call` method. If both are used, the output is a dictionary containing the results of both. NzAYou need to specify either an `audio` or `text` input to process.tokenizer_init_kwargs audio_kwargs text_kwargs input_idslabels) ValueError _merge_kwargsr r init_kwargsr)rrrr output_kwargsinputs encodingsrrr__call__ s&  zWav2Vec2Processor.__call__cOs|dd}|dd}t|dkr|d}|dd}|dur-|jj|g|Ri|}|dur;|jj|fi|}|durA|S|durG|S|d|d<|S)ag This method operates on batches of extracted features and/or tokenized text. It forwards all arguments to [`Wav2Vec2FeatureExtractor.pad`] and/or [`PreTrainedTokenizer.pad`] depending on the input modality and returns their outputs. If both modalities are passed, [`Wav2Vec2FeatureExtractor.pad`] and [`PreTrainedTokenizer.pad`] are called. Args: input_features: When the first argument is a dictionary containing a batch of tensors, or the `input_features` argument is present, it is passed to [`Wav2Vec2FeatureExtractor.pad`]. labels: When the `label` argument is present, it is passed to [`PreTrainedTokenizer.pad`]. Returns: This method returns the results of each `pad` method. If both are used, the output is a dictionary containing the results of both. input_featuresNr r)poplenrpadr)rargsrr(r rrrr-As     zWav2Vec2Processor.padcCs|jj}|dgS)Nr )rmodel_input_names)rfeature_extractor_input_namesrrrr/bs z#Wav2Vec2Processor.model_input_names)NN)r r r rrrstrlistrrrr r'r-propertyr/ __classcell__rrrrrs   !rN)__doc__processing_utilsrrrtokenization_utils_baserrrutilsrr r__all__rrrrs   M