U ,-e2@s$dZddlmZGdddeZdS)z$ Speech processor class for Whisper )ProcessorMixincsZeZdZdZdZdZfddZddd Zd d Zd d Z ddZ de dddZ Z 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. ZWhisperFeatureExtractorZWhisperTokenizercs t|||j|_d|_dS)NF)super__init__feature_extractorcurrent_processor_in_target_context_manager)selfr tokenizer __class__o/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/transformers/models/whisper/processing_whisper.pyr(szWhisperProcessor.__init__NTcCs|jj|||dS)N)tasklanguage no_timestamps)r get_decoder_prompt_ids)r rrrr r rr-sz'WhisperProcessor.get_decoder_prompt_idscOs|jr|j||S|dd}|dd}|dd}t|dkrV|d}|dd}|dkrn|dkrntd|dk r|j|f|d|i|}|dk r|j|f|}|dkr|S|dkr|S|d|d <|SdS) 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.Z input_idslabels)rrpoplen ValueErrorrr )r argskwargsrrrinputs encodingsr r r__call__0s(       zWhisperProcessor.__call__cOs|jj||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_decoder rrr r rr!RszWhisperProcessor.batch_decodecOs|jj||S)z This method forwards all its arguments to WhisperTokenizer's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )r decoder"r r rr#YszWhisperProcessor.decodenp)rcCs|jj||dS)N)return_tensors)r get_prompt_ids)r rr%r r rr&`szWhisperProcessor.get_prompt_ids)NNT)r$)__name__ __module__ __qualname____doc__Zfeature_extractor_classZtokenizer_classrrr r!r#strr& __classcell__r r r rrs   "rN)r*Zprocessing_utilsrrr r r rs