g22dZddlmZGddeZdS)z$ Speech processor class for Whisper )ProcessorMixincReZdZdZdZdZfdZddZdZd Z d Z dd e fd Z xZ S)WhisperProcessoraz Constructs a Whisper processor which wraps a Whisper feature extractor and a Whisper tokenizer into a single processor. [`WhisperProcessor`] offers all the functionalities of [`WhisperFeatureExtractor`] and [`WhisperTokenizer`]. See the [`~WhisperProcessor.__call__`] and [`~WhisperProcessor.decode`] for more information. Args: feature_extractor (`WhisperFeatureExtractor`): An instance of [`WhisperFeatureExtractor`]. The feature extractor is a required input. tokenizer (`WhisperTokenizer`): An instance of [`WhisperTokenizer`]. The tokenizer is a required input. WhisperFeatureExtractorWhisperTokenizercrt|||j|_d|_dS)NF)super__init__feature_extractorcurrent_processor_in_target_context_manager)selfr tokenizer __class__s j/var/www/html/ai-engine/env/lib/python3.11/site-packages/transformers/models/whisper/processing_whisper.pyr zWhisperProcessor.__init__(s7 *I666!%!7*/'''NTc<|j|||S)N)tasklanguage no_timestamps)rget_decoder_prompt_ids)rrrrs rrz'WhisperProcessor.get_decoder_prompt_ids-s ~44$an4ooorc|jr |j|i|S|dd}|dd}|dd}t|dkr|d}|dd}||t d||j|g|Rd|i|}||j|fi|}||S||S|d|d <|S) a  Forwards the `audio` argument to WhisperFeatureExtractor's [`~WhisperFeatureExtractor.__call__`] and the `text` argument to [`~WhisperTokenizer.__call__`]. Please refer to the doctsring of the above two methods for more information. audioN sampling_ratetextzAYou need to specify either an `audio` or `text` input to process. input_idslabels)r r poplen ValueErrorr r)rargskwargsrrrinputs encodingss r__call__zWhisperProcessor.__call__0s  * ;)4)4:6:: : 7D)) ?D99 zz&$'' t99q==GE8D =T\`aa a  +T+E`D``` `Y_``F  &t66v66I <M ] (5F8 Mrc&|jj|i|S)z This method forwards all its arguments to WhisperTokenizer's [`~PreTrainedTokenizer.batch_decode`]. Please refer to the docstring of this method for more information. )r batch_decoderr#r$s rr)zWhisperProcessor.batch_decodeRs +t~*D;F;;;rc&|jj|i|S)z This method forwards all its arguments to WhisperTokenizer's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )rdecoder*s rr,zWhisperProcessor.decodeYs %t~$d5f555rnprc:|j||S)N)return_tensors)rget_prompt_ids)rrr/s rr0zWhisperProcessor.get_prompt_ids`s~,,T.,QQQr)NNT)r-)__name__ __module__ __qualname____doc__feature_extractor_classtokenizer_classr rr'r)r,strr0 __classcell__)rs@rrrs  8(O00000 pppp   D<<<666RR3RRRRRRRRrrN)r4processing_utilsrrrrr;sk/.....KRKRKRKRKR~KRKRKRKRKRr