1i |dZddlmZmZmZddlmZmZmZddl m Z GddedZ e Gd d eZ d gZ y ) z% Speech processor class for Wav2Vec2 )ProcessingKwargsProcessorMixinUnpack) AudioInputPreTokenizedInput TextInput)auto_docstringceZdZiZy)Wav2Vec2ProcessorKwargsN)__name__ __module__ __qualname__ _defaultsl/mnt/e/genesis-system/.venv/lib/python3.12/site-packages/transformers/models/wav2vec2/processing_wav2vec2.pyr r sIrr F)totalc eZdZfdZe d dedzdeeezeze zdzde e fdZ dZ edZxZS) Wav2Vec2Processorc&t|||y)N)super__init__)selffeature_extractor tokenizer __class__s rrzWav2Vec2Processor.__init__s *I6rNaudiotextkwargsc | | td|jtfd|jji|}||j |fi|d}||j|fi|d}|S|Sdd<|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. zAYou 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 encodingss r__call__zWav2Vec2Processor.__call__ s =T\`a a*** # "&.."<"<   +T++ES]>5RSF  &tL}]/KLI <M ] (5F8 Mrc |jdd}|jdd}t|dkDr |d}|dd}| |jj|g|i|}||jj|fi|}||S||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%s rr3zWav2Vec2Processor.padAs $4d;Hd+ t9q=!!WN8D  %7T3377XXQWXN  'T^^''9&9F >! !  #M'-k':N8 $! !rc:|jj}|dgzS)Nr%)rmodel_input_names)rfeature_extractor_input_namess rr6z#Wav2Vec2Processor.model_input_namesbs#)-(>(>(P(P%,z99r)NN)r r rrr rstrlistrrrr r,r3propertyr6 __classcell__)rs@rrrs~7$(GKD DIo ),==D01 @"B::rrN)__doc__processing_utilsrrrtokenization_utils_baserrrutilsr r r__all__rrrrAsTIHOO#.eJ:J:J:Z  r