o Zh2@s*dZddlmZGdddeZdgZdS)z Processor class for CLVP )ProcessorMixincsHeZdZdZdZdZgdZfddZddZd d Z d d Z Z S) ClvpProcessoras Constructs a CLVP processor which wraps a CLVP Feature Extractor and a CLVP Tokenizer into a single processor. [`ClvpProcessor`] offers all the functionalities of [`ClvpFeatureExtractor`] and [`ClvpTokenizer`]. See the [`~ClvpProcessor.__call__`], [`~ClvpProcessor.decode`] and [`~ClvpProcessor.batch_decode`] for more information. Args: feature_extractor (`ClvpFeatureExtractor`): An instance of [`ClvpFeatureExtractor`]. The feature extractor is a required input. tokenizer (`ClvpTokenizer`): An instance of [`ClvpTokenizer`]. The tokenizer is a required input. ZClvpFeatureExtractorZ ClvpTokenizer) input_idsZinput_featuresattention_maskcst||dS)N)super__init__)selffeature_extractor tokenizer __class__W/var/www/auris/lib/python3.10/site-packages/transformers/models/clvp/processing_clvp.pyr-szClvpProcessor.__init__cOs|dd}|dd}|dd}|dur|durtd|dur-|j|fd|i|}|dur:|j|fi|}|dur@|S|durF|S|d|d<|d|d<|S)z Forwards the `audio` and `sampling_rate` arguments to [`~ClvpFeatureExtractor.__call__`] and the `text` argument to [`~ClvpTokenizer.__call__`]. Please refer to the docstring of the above two methods for more information. raw_speechN sampling_ratetextzFYou need to specify either an `raw_speech` or `text` input to process.rr)pop ValueErrorr r )rargskwargsrrrZinputs encodingsr r r__call__0s     zClvpProcessor.__call__cO|jj|i|S)z This method forwards all its arguments to ClvpTokenizer's [`~PreTrainedTokenizer.batch_decode`]. Please refer to the docstring of this method for more information. )r batch_decoderrrr r rrMzClvpProcessor.batch_decodecOr)z This method forwards all its arguments to ClvpTokenizer's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )r decoderr r rrUrzClvpProcessor.decode) __name__ __module__ __qualname____doc__Zfeature_extractor_classZtokenizer_classZmodel_input_namesrrrr __classcell__r r r rrs  rN)r Zprocessing_utilsrr__all__r r r rs  F