o ZhS@s*dZddlmZGdddeZdgZdS)z$ Speech processor class for Whisper )ProcessorMixincsZeZdZdZdZdZfddZddd Zd d Zd d Z ddZ dde fddZ 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__]/var/www/auris/lib/python3.10/site-packages/transformers/models/whisper/processing_whisper.pyr(s zWhisperProcessor.__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|i|S|dd}|dd}|dd}t|dkr-|d}|dd}|dur9|dur9td|durK|j|g|Rd|i|}|durX|j|fi|}|dur^|S|durd|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 docstring 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 argskwargsrrrZinputs encodingsr r r__call__0s(      zWhisperProcessor.__call__cO|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_decoder rrr r rr!RzWhisperProcessor.batch_decodecOr )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$Yr#zWhisperProcessor.decodenprcCs|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_classrrrr!r$strr' __classcell__r r r rrs  "rN)r+Zprocessing_utilsrr__all__r r r rs  N