ZjSrSSKJr SSKJr SSKJr SSKJrJ r \ R"\ 5r \"SS9\"S S \555r \"SS9\"S S \555r\"SS9\"S S\555r/SQrg)zDia model configuration)strict)PreTrainedConfig)RopeParameters)auto_docstringloggingznari-labs/Dia-1.6B) checkpointc\rSrSr%SrSr\\S'Sr\\S'Sr \\S'Sr \\S 'Sr \\S 'S r \\S 'S r \\S'Sr\\S'Sr\\S'Sr\\S'Sr\S-\S'Sr\\S'Srg)DiaEncoderConfig dia_encodermax_position_embeddings num_hidden_layers hidden_sizenum_attention_headsnum_key_value_headshead_dimiintermediate_sizeh㈵>norm_eps vocab_sizesilu hidden_actNrope_parameters{Gz?initializer_range)__name__ __module__ __qualname____firstlineno__ model_typerint__annotations__rrrrrrrfloatrrstrrdictr!__static_attributes__r"z/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/models/dia/configuration_dia.pyr r sJ#'S'sK!!!!Hc!s!HeJJ#'OTD['#u#r.r c|\rSrSr%SrSrSr\\S'Sr \\S'Sr \\S 'S r \\S 'S r \\S 'Sr \\S'Sr\\S'S r\\S'Sr\\S'S r\\S'Sr\\S'Sr\\S'Sr\\S'Sr\\S'Sr\\S'Sr\\-S-\S 'S!r\\S"'S#r\\S$'S#r\\S%'S&r \S-\S''Sr!\\"\-S-\S('S)r#\S-\S*'S+r$g),DiaDecoderConfig-a cross_num_attention_heads (`int`, *optional*, defaults to 16): Number of attention heads for each cross-attention layer in the Transformer decoder. cross_head_dim (`int`, *optional*, defaults to 128): Dimensionality of the cross-attention head. cross_num_key_value_heads (`int`, *optional*, defaults to 16): Number of key and value heads for each cross-attention layer in the Transformer decoder. cross_hidden_size (`int`, *optional*, defaults to 1024): Dimensionality of the cross-attention layers. dia_decoderi rriri rrrrrrcross_num_attention_headscross_head_dimcross_num_key_value_headsrcross_hidden_sizerrirrr num_channelsNrr r!T use_cacheis_encoder_decoderi pad_token_id eos_token_idi bos_token_idr")%r#r$r%r&__doc__r'rr(r)rrrrrrr6r7r8r9rr*rrr+r;rrr,r!r<boolr=r>r?listr@r-r"r.r/r1r1-s J#'S'sK!s!!!  Hc%'s'NC%'s'!s!HeJJL#48O^d*T18#u#It###L#*#+/L#S /D(/#L#*#r.r1c^\rSrSr%SrSrS/r\\S.r Sr \\ -S-\ S'Sr \\ -S-\ S'S r\\ S 'S r\\ S 'Sr\S-\ S 'Sr\\\-S-\ S'Sr\S-\ S'Sr\\S-\ S'Sr\\ S'S r\\ S'U4SjrSrSrSrU=r$) DiaConfigUa delay_pattern (`list[int]`, *optional*, defaults to `[0, 8, 9, 10, 11, 12, 13, 14, 15]`): The delay pattern for the decoder. The length of this list must match `decoder_config.num_channels`. Example: ```python >>> from transformers import DiaConfig, DiaModel >>> # Initializing a DiaConfig with default values >>> configuration = DiaConfig() >>> # Initializing a DiaModel (with random weights) from the configuration >>> model = DiaModel(configuration) >>> # Accessing the model configuration >>> configuration = model.config ``` diapast_key_values)encoder_configdecoder_configNrIrJrrTr=r>r?r@ delay_patternr r!r<c \>[UR[5(a[S0URD6Ul[UR[5(a[ S0URD6UlURb URO [5UlURb URO [ 5UlUR b UR O/SQUlURb0[RS5 URURlURb0[RS5 URURl URb0[RS5 URURl [TU]4"S0UD6 g)N) rr: r zjPassing `pad_token_id` to `DiaConfig` is deprecated. Please set it directly on `DiaDecoderConfig` instead.zjPassing `eos_token_id` to `DiaConfig` is deprecated. Please set it directly on `DiaDecoderConfig` instead.zjPassing `bos_token_id` to `DiaConfig` is deprecated. Please set it directly on `DiaDecoderConfig` instead.r") isinstancerIr,r rJr1rKr>logger warning_oncer?r@super __post_init__)selfkwargs __class__s r/rWDiaConfig.__post_init__{sf d))4 0 0"2"IT5H5H"ID  d))4 0 0"2"IT5H5H"ID 595H5H5Td11ZjZl595H5H5Td11ZjZl"&"4"4"@D  Fg     (   H 04/@/@D   ,    (   H 04/@/@D   ,    (   H 04/@/@D   , ''r.ctURR[UR5:wa [ S5eg)zOPart of `@strict`-powered validation. Validates the architecture of the config.z3Number of channels must match delay pattern length.N)rJr;lenrK ValueError)rXs r/validate_architectureDiaConfig.validate_architectures3    + +s43E3E/F FRS S Gr.cUR$)z^Defaulting to audio config as it's the decoder in this case which is usually the text backbone)rJ)rXargsrYs r/get_text_configDiaConfig.get_text_configs"""r.)rJrKrI)r#r$r%r&rAr'keys_to_ignore_at_inferencer r1 sub_configsrIr,r)rJrr*r=rBr>r(r?rCr@rKr!r<rWr_rcr- __classcell__)rZs@r/rErEUs(J#4"5%5IYZK59N$t+d2959N$t+d29He###L#*#+/L#S /D(/#L#*#&*M49t#*#u#It"(HT ##r.rE)rEr r1N)rAhuggingface_hub.dataclassesrconfiguration_utilsrmodeling_rope_utilsrutilsrr get_loggerr#rTr r1rE__all__r"r.r/rns.31,   H %/0$'$1$"/0#$'#$1#$L/0O# O#1O#d @r.