ihddlmZmZddlmZmZmZddlmZm Z ddl m Z m Z e jeZGddedZe Gd d eZd gZy ) ) AudioInputmake_list_of_audio)ProcessingKwargsProcessorMixinUnpack)PreTokenizedInput TextInput)auto_docstringloggingc,eZdZdddddddddd id Zy ) ParakeetProcessorKwargsi>longestT) sampling_ratepaddingreturn_attention_maskrightF)r padding_sideadd_special_tokensreturn_tensorspt) audio_kwargs text_kwargs common_kwargsN)__name__ __module__ __qualname__ _defaultsl/mnt/e/genesis-system/.venv/lib/python3.12/site-packages/transformers/models/parakeet/processing_parakeet.pyr r s3# %) #"' +D1 Irr F)totalc eZdZfdZe d dedeezeezeezdzde dzde e fdZ e dZxZS) ParakeetProcessorc&t|||y)N)super__init__)selffeature_extractor tokenizer __class__s r r&zParakeetProcessor.__init__*s *I6rNaudiotextrkwargsc xt|}|jtfd|jji|}| t j d|dddn#||ddk7rtd|d|ddd||j|fi|d}||j|fi|d }|Sd d <|S) a sampling_rate (`int`, *optional*): The sampling rate of the input audio in Hz. This should match the sampling rate expected by the feature extractor (defaults to 16000 Hz). If provided, it will be validated against the processor's expected sampling rate, and an error will be raised if they don't match. If not provided, a warning will be issued and the default sampling rate will be assumed. tokenizer_init_kwargszUYou've provided audio without specifying the sampling rate. It will be assumed to be rrz$, which can result in silent errors.z The sampling rate of the audio (z5) does not match the sampling rate of the processor (zD). Please provide resampled the audio to the expected sampling rate.r input_idslabels) r _merge_kwargsr r) init_kwargslogger warning_once ValueErrorr()r'r+r,rr- output_kwargsinputs encodingss r __call__zParakeetProcessor.__call__-sO#5)*** # "&.."<"<     ghuwEiFGViWhXX|} mN;OL L2=/AvxEFTxUVexfwggkl   +T++ES]>5RSF  &tL}]/KLI <M(5F8 Mrc:|jj}|dgzS)Nr1)r(model_input_names)r'feature_extractor_input_namess r r<z#ParakeetProcessor.model_input_namesXs!(,(>(>(P(P%,z99r)NN)rrrr&r rr rlistintrr r:propertyr< __classcell__)r*s@r r#r#(s7bf$( ((++d9o=EV@WWZ^^(Tz ( 01 ((T::rr#N) audio_utilsrrprocessing_utilsrrrtokenization_utils_baserr utilsr r get_loggerrr4r r#__all__rrr rHs_:HHC,   H % .e  2:2:2:j  r