sgdZddlmZmZmZddlZddlmZddl m Z ddl m Z m Z mZejeZGdd eZy) zFeature extractor class for DAC)ListOptionalUnionN)SequenceFeatureExtractor) BatchFeature)PaddingStrategy TensorTypeloggingceZdZdZddgZ ddedededeffd Z dd ee je ee e je e efd e ee eefd e e d e ed e eeefde edefdZxZS)DacFeatureExtractora> Constructs an Dac 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 :t|d|||d|||_y)N)rrr)super__init__r)selfrrrrkwargs __class__s a/var/www/html/venv/lib/python3.12/site-packages/transformers/models/dac/feature_extraction_dac.pyrzDacFeatureExtractor.__init__1s' wl-_lwpvw$ raw_audiopadding truncation max_lengthreturn_tensorsreturnc |;||jk7rAtd|d|jd|jd|d tjd|r |r td|d }t t |t tfxr(t |d tjtt f}|r=|Dcgc]1}tj|tj j3}}n|s@t |tjs&tj|tj }nht |tjrN|jtjtjur|jtj}|s tj|jg}t!|D]\} } | j"d kDrtd | j$|j&dk(r+| j"dk7rtd| j$dd|j&d k(swtdt)d|i} |j+| |||d|j,} |r)| j.ddtj0ddf| _g} | j3dD]1} |j&dk(r| d} | j5| j3| | d<|| j7|} | Scc}w)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. Nz3The 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 .zIt is strongly recommended to pass the `sampling_rate` argument to this function. 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$Stereo audio isn't supported for nowrF)r rrreturn_attention_maskpad_to_multiple_of).N)r ValueErrorloggerwarningbool isinstancelisttuplenpndarrayasarrayfloat32Tr%float64astype enumeratendimshaperrpadrrnewaxispopappendconvert_to_tensors) rrrrr r!r is_batchedaudioidxexampler padded_inputss r__call__zDacFeatureExtractor.__call__<sT  $ 2 22 I$P**+,**+9]O1F NN\  z`a a _G y4- 0 jj1PRPZPZ\acgOh6i  LUV5E<>>VIVJy"**$E 9BJJ?I  2:: .9??bhhrzzFZ3Z!((4II.001I&i0 ILC||a #YZaZgZgYh!ijj  A%',,!*; #G VXHYGZZc!dee  A% !GHH  I$^Y$?@  !!"'# !  )6)C)CArzzSTDT)UM & $((8 +G  A%!),    * + )5 n%  %)<<^LMYWs56L)r'i>gi)NFNNN)__name__ __module__ __qualname____doc__model_input_namesintfloatrrr2r3rrr.strr r rrF __classcell__)rs@rr r s"(8"" % % % %  %@D%*$(;?'+nT%[$rzz2BDeDUUVn%c? :;<nTN n SM n !sJ!78 n }n nrr )rJtypingrrrnumpyr2!feature_extraction_sequence_utilsrfeature_extraction_utilsrutilsr r r get_loggerrGr,r rrrrVsB&((I499   H %M2Mr