T^(hS0dZddlmZGddeZdgZy)z$ Speech processor class for Whisper )ProcessorMixincReZdZdZdZdZfdZd dZdZdZ dZ d d 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. WhisperFeatureExtractorWhisperTokenizercVt||||j|_d|_y)NF)super__init__feature_extractorcurrent_processor_in_target_context_manager)selfr tokenizer __class__s l/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/transformers/models/whisper/processing_whisper.pyr zWhisperProcessor.__init__(s) *I6!%!7!7*/'c>|jj|||S)N)tasklanguage no_timestamps)rget_decoder_prompt_ids)rrrrs rrz'WhisperProcessor.get_decoder_prompt_ids-s~~44$an4oorcz|jr|j|i|S|jdd}|jdd}|jdd}t|dkDr |d}|dd}| | t d||j |g|d|i|}||j |fi|}|S|Sdd <|S) a  Forwards the `audio` argument to WhisperFeatureExtractor's [`~WhisperFeatureExtractor.__call__`] and the `text` argument to [`~WhisperTokenizer.__call__`]. Please refer to the docstring 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) ?D9 zz&$' t9q=GE8D =T\`a a  +T++E`D` `Y_`F  &t6v6I <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~~$$d5f55rrc<|jj||S)N)return_tensors)rget_prompt_ids)rrr.s rr/zWhisperProcessor.get_prompt_ids`s~~,,T.,QQr)NNT)np)__name__ __module__ __qualname____doc__feature_extractor_classtokenizer_classr rr'r)r,strr/ __classcell__)rs@rrrs> 8(O0 p D<6R3RrrN)r4processing_utilsrr__all__rrr<s*/KR~KR\  r