cCi5!SrSSKJrJr SSKrSSKJr SSKJ r SSK J r J r J r \ R"\5r"SS \5rS /rg) zFeature extractor class for Dia)OptionalUnionN)SequenceFeatureExtractor) BatchFeature)PaddingStrategy TensorTypeloggingc ^\rSrSrSrSS/rSS\S\S\S\4U4S jjjrSS \ \ R\ \\ \ R\ \ \4S \ \ \\\4S \ \S \ \S\ \ \\4S\ \S\4SjjrSrU=r$)DiaFeatureExtractora  Constructs an Dia feature extractor. This feature extractor inherits from [`~feature_extraction_sequence_utils.SequenceFeatureExtractor`] which contains most of the main methods. Users should refer to this superclass for more information regarding those methods. Args: feature_size (`int`, *optional*, defaults to 1): The feature dimension of the extracted features. Use 1 for mono, 2 for stereo. sampling_rate (`int`, *optional*, defaults to 16000): The sampling rate at which the audio waveform should be digitalized, expressed in hertz (Hz). padding_value (`float`, *optional*, defaults to 0.0): The value that is used for padding. hop_length (`int`, *optional*, defaults to 512): Overlap length between successive windows. input_values n_quantizers feature_size sampling_rate padding_value hop_lengthc 8>[TU]"SXUS.UD6 X@lg)N)rrr)super__init__r)selfrrrrkwargs __class__s h/home/james-whalen/.local/lib/python3.13/site-packages/transformers/models/dia/feature_extraction_dia.pyrDiaFeatureExtractor.__init__1s# wl_lwpvw$ raw_audiopadding truncation max_lengthreturn_tensorsreturnc Ub<X`R:wa,[SUSURSURSUS3 5eO-[RSURR S35 U(aU(a [S5eUcS n[ [U[[45=(a( [US [R[[455nU(a>UVs/sH0n[R"U[RS 9RPM2 nnOU(dC[U[R5(d$[R"U[RS 9nOo[U[R5(aPUR[R"[R 5LaUR#[R5nU(d![R"U5R/n[%U5HBupUR&S :XdMU R(S :XdM)[R*"U S 5X'MD [%U5HupU R(S :a[SU R,35eUR&S:Xa,U R(S:wa[SU R,S S35eUR&S :XdM{U R(S:wdM[SU R,S S35e [/SU05n UR&n SUlUR1U UUUS UR2S9n U R5S5U S'/n U R5S5H3n UR&S:XaU Sn U R7U R5 M5 XS'UbU R9U5n XlU $s snf)a Main method to featurize and prepare for the model one or several sequence(s). Args: raw_audio (`np.ndarray`, `list[float]`, `list[np.ndarray]`, `list[list[float]]`): The sequence or batch of sequences to be processed. Each sequence can be a numpy array, a list of float values, a list of numpy arrays or a list of list of float values. The numpy array must be of shape `(num_samples,)` for mono audio (`feature_size = 1`), or `(2, num_samples)` for stereo audio (`feature_size = 2`). padding (`bool`, `str` or [`~utils.PaddingStrategy`], *optional*, defaults to `True`): Select a strategy to pad the returned sequences (according to the model's padding side and padding index) among: - `True` or `'longest'`: Pad to the longest sequence in the batch (or no padding if only a single sequence if provided). - `'max_length'`: Pad to a maximum length specified with the argument `max_length` or to the maximum acceptable input length for the model if that argument is not provided. - `False` or `'do_not_pad'` (default): No padding (i.e., can output a batch with sequences of different lengths). truncation (`bool`, *optional*, defaults to `False`): Activates truncation to cut input sequences longer than `max_length` to `max_length`. max_length (`int`, *optional*): Maximum length of the returned list and optionally padding length (see above). return_tensors (`str` or [`~utils.TensorType`], *optional*, default to 'pt'): If set, will return tensors instead of list of python integers. Acceptable values are: - `'tf'`: Return TensorFlow `tf.constant` objects. - `'pt'`: Return PyTorch `torch.Tensor` objects. - `'np'`: Return Numpy `np.ndarray` objects. sampling_rate (`int`, *optional*): The sampling rate at which the `audio` input was sampled. It is strongly recommended to pass `sampling_rate` at the forward call to prevent silent errors. z3The model corresponding to this feature extractor: z& was trained using a sampling rate of zB. Please make sure that the provided audio input was sampled with z and not .zDIt is strongly recommended to pass the `sampling_rate` argument to `zN()`. Failing to do so can result in silent errors that might be hard to debug.zABoth padding and truncation were set. Make sure you only set one.Tr)dtypez6Expected input shape (channels, length) but got shape z$Expected mono audio but example has z channelsz&Expected stereo audio but example has r)r!r rreturn_attention_maskpad_to_multiple_ofattention_mask padding_mask).N)r ValueErrorloggerwarningr__name__bool isinstancelisttuplenpndarrayasarrayfloat32Tr&float64astype enumeraterndimmeanshaperpadrpopappendconvert_to_tensors)rrrr r!r"r is_batchedaudioidxexampleroriginal_feature_size padded_inputss r__call__DiaFeatureExtractor.__call__<seT  $ 2 22 I$P**+,**+9]O1F3 NNVW[WeWeWnWnVop\\  z`a a _G y4- 0 jj1PRPZPZ\acgOh6i  LUVI5E<>>IIVIJy"**$E$E 9BJJ?I  2:: . .9??bhhrzzFZ3Z!((4II.001I&i0LC  A%',,!*;!#"!5 1 &i0LC||a #YZaZgZgYh!ijj  A%',,!*; #G VXHYGZZc!dee  A%',,!*; #I'--XZJ[I\\e!fgg 1$^Y$?@ !% 1 1 !!"&# ! )6(9(9:J(K n% $((8G  A%!),    *9 )5n%  %)<<^LM2mWs$7O)rr)r)i>gi)NFNNN)r1 __module__ __qualname____firstlineno____doc__model_input_namesintfloatrrr6r7r4rr2strrr rrK__static_attributes__ __classcell__)rs@rr r s"(8"" % % % %  % %@D%*$(;?'+xT%[$rzz2BDeDUUVx%c? :;<xTN x SM x !sJ!78 x }x xxrr )rPtypingrrnumpyr6!feature_extraction_sequence_utilsrfeature_extraction_utilsrutilsrr r get_loggerr1r/r __all__rrrr^sH&"I499   H %W2Wt ! !r