tj dZddlmZmZmZddlmZmZmZddl m Z GddedZ e Gd d eZ d gZ d S) z% Speech processor class for Wav2Vec2 )ProcessingKwargsProcessorMixinUnpack) AudioInputPreTokenizedInput TextInput)auto_docstringceZdZiZdS)Wav2Vec2ProcessorKwargsN)__name__ __module__ __qualname__ _defaultsn/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/transformers/models/wav2vec2/processing_wav2vec2.pyr r sIIIrr F)totalc eZdZfdZe d dedzdeeezeze zdzde e fdZ dZ edZxZS) Wav2Vec2ProcessorcLt||dS)N)super__init__)selffeature_extractor tokenizer __class__s rrzWav2Vec2Processor.__init__s$ *I66666rNaudiotextkwargsc ||td|jtfd|jji|}||j|fi|d}||j|fi|d}||S||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 encodingss r__call__zWav2Vec2Processor.__call__ s =T\`aa a** #  "&."<    +T+ESS]>5RSSF  &tLL}]/KLLI <M ] (5F8 Mrc(|dd}|dd}t|dkr|d}|dd}||jj|g|Ri|}||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++ t99q==!!WN8D  %7T37XXXXQWXXN  'T^'99&99F >! !  #M'-k':N8 $! !rc&|jj}|dgzS)Nr%)rmodel_input_names)rfeature_extractor_input_namess rr6z#Wav2Vec2Processor.model_input_namesbs)-(>(P%,z99r)NN)r r rrr rstrlistrrrr r,r3propertyr6 __classcell__)rs@rrrs77777$(GKD DIo ),==D01 ^@"""B::X:::::rrN)__doc__processing_utilsrrrtokenization_utils_baserrrutilsr r r__all__rrrrAsIHHHHHHHHHOOOOOOOOOO######.eJ:J:J:J:J:J:J:J:Z  r