o ei@s:dZddlmZddlmZeGdddeZdgZdS)z$Speech processor class for SpeechT5.)ProcessorMixin)auto_docstringcs0eZdZfddZeddZddZZS)SpeechT5Processorcst||dS)N)super__init__)selffeature_extractor tokenizer __class__n/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/transformers/models/speecht5/processing_speecht5.pyrszSpeechT5Processor.__init__c Osd|dd}|dd}|dd}|dd}|dd}|dur*|dur*td|dur6|dur6td|durJ|durJ|durJ|durJtd|dur]|j|g|Rd|i|}n|durk|j|fi|}nd}|dur|j|||d |} | d } n|dur|j|fi|} | d } nd} |dur| S| dur| |d <| d } | dur| |d<|S)Naudiotext text_target audio_target sampling_ratez\Cannot process both `audio` and `text` inputs. Did you mean `audio_target` or `text_target`?z\Cannot process both `audio_target` and `text_target` inputs. Did you mean `audio` or `text`?zaYou need to specify either an `audio`, `audio_target`, `text`, or `text_target` input to process.)rr input_values input_idslabelsattention_maskdecoder_attention_mask)pop ValueErrorrr get) rargskwargsrrrrrinputstargetsrrr r r __call__sJ         zSpeechT5Processor.__call__c Osd|dd}|dd}|dd}|dur|durtd|dur.|dur.|dur.td|dur@|jj|g|Ri|}n|durO|jj|fi|}nd}|durd|vsdt|trsd|dvrs|jj|fi|}|d}n"|jj}|jj|j_|jj|g|Ri|}||j_|d}nd}|dur|S|dur||d<| d} | dur| |d <|S) au Collates the audio and text inputs, as well as their targets, into a padded batch. Audio inputs are padded by SpeechT5FeatureExtractor's [`~SpeechT5FeatureExtractor.pad`]. Text inputs are padded by SpeechT5Tokenizer's [`~SpeechT5Tokenizer.pad`]. Valid input combinations are: - `input_ids` only - `input_values` only - `labels` only, either log-mel spectrograms or text tokens - `input_ids` and log-mel spectrogram `labels` - `input_values` and text `labels` Please refer to the docstring of the above two methods for more information. rNrrz:Cannot process both `input_values` and `input_ids` inputs.zZYou need to specify either an `input_values`, `input_ids`, or `labels` input to be padded.rr) rrrpadr isinstancelist feature_size num_mel_binsr) rrrrrrrrfeature_size_hackrr r r r!Js@       zSpeechT5Processor.pad)__name__ __module__ __qualname__rrrr! __classcell__r r r r rs   0rN)__doc__processing_utilsrutilsrr__all__r r r r s    r