a h@sxdZddlZddlmZmZddlZddlmZm Z m Z m Z ddl m Z ddlmZeeZGdd d e Zd gZdS) zXcodec model configurationN)OptionalUnion) AutoConfig DacConfig HubertConfig WavLMConfig)PretrainedConfig)loggingcseZdZdZdZeedZdddddgddgddgdddd ddf ee e e e e e e e e e e e ee e e e efe e efd fd d Zee d ddZee d ddZee d ddZee d ddZee d ddZee d ddZZS) XcodecConfiga This is the configuration class to store the configuration of an [`XcodecModel`]. It is used to instantiate a Xcodec model according to the specified arguments, defining the model architecture. Instantiating a configuration with the defaults will yield a similar configuration to that of the [Manel/X-Codec](https://huggingface.co/Manel/X-Codec) architecture. Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the documentation from [`PretrainedConfig`] for more information. Args: target_bandwidths (`List[float]`, *optional*, defaults to `[0.5, 1, 1.5, 2, 4]`): The range of different bandwidths (in kbps) the model can encode audio with. sample_rate (`int`, *optional*, defaults to 16000): The sampling rate at which the audio waveform should be digitalized, in hertz (Hz). kernel_size (`int`, *optional*, defaults to 3): Kernel size for the initial semantic convolution. channel_ratios (`List[float]`, *optional*, defaults to `[1, 1]`): Expansion factors for the number of output channels in each semantic block. strides (`List[int]`, *optional*, defaults to `[1, 1]`): Strides for each semantic encoder block. block_dilations (`List[int]`, *optional*, defaults to `[1, 1]`): Dilation factors for the residual units in semantic blocks. unit_kernel_size (`int`, *optional*, defaults to 3): Kernel size inside each ResidualUnit in semantic blocks. codebook_size (`int`, *optional*, defaults to 1024): Number of entries in each residual quantizer's codebook. codebook_dim (`int`, *optional*): Dimensionality of each codebook vector. Defaults to sum of hidden size of acoustic and semantic models. initializer_range (`float`, *optional*, defaults to 0.02): Standard deviation of the truncated normal initializer for all weight matrices. acoustic_model_config (`Union[Dict, DacConfig]`, *optional*): An instance of the configuration for the acoustic (DAC) model. semantic_model_config (`Union[Dict, HubertConfig, WavLMConfig]`, *optional*): An instance of the configuration object for the semantic (HuBERT) model. Example: ```python >>> from transformers import XcodecModel, XcodecConfig >>> # Initializing configuration >>> configuration = XcodecConfig() >>> # Initializing a model (with random weights) from the configuration >>> model = XcodecModel(configuration) >>> # Accessing the model configuration >>> configuration = model.config ```Zxcodec)acoustic_model_configsemantic_model_configNi>rg{Gz?) target_bandwidths sample_rate kernel_sizechannel_ratiosstridesblock_dilationsunit_kernel_size codebook_size codebook_diminitializer_ranger r c  sdtjfi| | dur8tdgddgddd|_n@t| trTtfi| |_n$t| trf| |_ntdt| | durt|_ nnt| trd| vrt | d|_ qt dtfi| |_ n.t| tst| tr| |_ ntd t| |dur gd }||_||_||_||_||_||_||_||_| |_| durZ|jj|j j} | |_dS) N@)r)Zencoder_hidden_sizedownsampling_ratiosZdecoder_hidden_sizeZupsampling_ratios hidden_sizezDacoustic_model_config must be a dict or DacConfig instance, but got Z _name_or_pathz_Could not determine semantic model type from config architecture. Defaulting to `HubertConfig`.zUsemantic_model_config must be a dict, HubertConfig, or WavLMConfig instance, but got )g?rg?rr)super__init__rr isinstancedict ValueErrortyperr rZfrom_pretrainedloggerwarningrrrrrrrrrrr!r)selfrrrrrrrrrrr r kwargs __class__k/var/www/html/assistant/venv/lib/python3.9/site-packages/transformers/models/xcodec/configuration_xcodec.pyr#YsX        zXcodecConfig.__init__)returncCst|j|jSN)mathceilr hop_lengthr*r.r.r/ frame_rateszXcodecConfig.frame_ratecCs|jjSr1)r r!r5r.r.r/semantic_hidden_sizesz!XcodecConfig.semantic_hidden_sizecCstt|jjSr1)intnpprodr r r5r.r.r/r4szXcodecConfig.hop_lengthcCstt|jSr1)r2r3log2rr5r.r.r/codebook_nbitsszXcodecConfig.codebook_nbitscCs|jj|jjSr1)r r!r r5r.r.r/r!szXcodecConfig.hidden_sizecCstd|jd|j|jS)Ni)r8rr6r<r5r.r.r/num_quantizersszXcodecConfig.num_quantizers)__name__ __module__ __qualname____doc__Z model_typerrZ sub_configsrlistfloatr8rr%rr#propertyr6r7r4r<r!r> __classcell__r.r.r,r/r sV2   Hr )rBr2typingrrnumpyr9Z transformersrrrrZconfiguration_utilsr utilsr Z get_loggerr?r(r __all__r.r.r.r/s