+ jRt^RIHt^RIHt^RIHt^RIHtH t ] P!] 4t ]!RR7]!RR]444t ]!RR7]!R R ]444t]!RR7]!R R ]444t.ROtR #)zDia model configuration)strict)PreTrainedConfig)RopeParameters)auto_docstringloggingznari-labs/Dia-1.6B) checkpointc`a]tRt^toRtRt^ tRt^t^t ^t Rt Rt Rt RtRtRtV3R ltR tVtR#) DiaEncoderConfig dia_encoderih㈵>siluN{Gz?c<V^8dQh/S[;R&S[;R&S[;R&S[;R&S[;R&S[;R&S[;R&S[;R&S[;R &S[;R &S[R ,;R &S[;R &#)max_position_embeddingsnum_hidden_layers hidden_sizenum_attention_headsnum_key_value_headshead_dimintermediate_sizenorm_eps vocab_size hidden_actNrope_parametersinitializer_range)intfloatstrdict)format __classdict__s"j/home/luca/screen-checking/.venv/lib/python3.14/site-packages/transformers/models/dia/configuration_dia.py __annotate__DiaEncoderConfig.__annotate__s !'   !!!D[' #!)__name__ __module__ __qualname____firstlineno__ model_typerrrrrrrrrrrr__annotate_func____static_attributes____classdictcell__r#s@r$r r sUJ#'K!!H!HJJ#'O#!r'r ca]tRt^-toRtRtRt^tRtRt ^t ^t ^t ^t ^t^tRtRtRtR t^ tR tR tR tR tR tRtRtV3RltRtVtR #)DiaDecoderConfiga 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 ii r r irNrTiic<V^8dQh/S[;R&S[;R&S[;R&S[;R&S[;R&S[;R&S[;R&S[;R&S[;R &S[;R &S[;R &S[;R &S[;R &S[;R&S[;R&S[S[,R,;R&S[;R&S[;R&S[;R&S[R,;R&S[S[S[,,R,;R&S[R,;R&#)rrrrrrrrcross_num_attention_headscross_head_dimcross_num_key_value_headscross_hidden_sizerrr num_channelsNrr use_cacheis_encoder_decoder pad_token_id eos_token_id bos_token_id)rrr rr!boollist)r"r#s"r$r%DiaDecoderConfig.__annotate__-sN !'!"#$%&!'(!)* +,-. #'/012 #'34!56789:;<=>$d*T18?@#ABCD#EF*#GHS /D(/IJ*#Kr'r()r)r*r+r,__doc__r-rrrrrrrr6r7r8r9rrrr:rrr;r<r=r>r?r.r/r0r1s@r$r3r3-s J#'K!! H%'N%'!HJJL48O#I##L+/L#LKr'r3caa]tRt^UtoRtRtR.tR]R]/t Rt Rt Rt Rt RtRtRtRtR tRtV3R ltR tR tV3R ltRtVtV;t#) DiaConfiga 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_valuesencoder_configdecoder_configNr Trc n<\VP\4'd\R/VPBVn\VP\4'd\ R/VPBVnVPe VPM \4VnVPe VPM \ 4VnVP e VP M.ROVnVPe1\PR4VPVPnVPe1\PR4VPVPn VPe1\PR4VPVPn \SV`4!R/VBR#)NzjPassing `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()  ) isinstancerHr!r rIr3 delay_patternr=logger warning_oncer>r?super __post_init__)selfkwargs __class__s&,r$rYDiaConfig.__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'c xVPP\VP48wd \ R4hR#)zOPart of `@strict`-powered validation. Validates the architecture of the config.z3Number of channels must match delay pattern length.N)rIr:lenrU ValueError)rZs&r$validate_architectureDiaConfig.validate_architectures3    + +s43E3E/F FRS S Gr'c VP#)z^Defaulting to audio config as it's the decoder in this case which is usually the text backbone)rI)rZargsr[s&*,r$get_text_configDiaConfig.get_text_configs"""r'c6<V^8dQh/S[S[,R,;R&S[S[,R,;R&S[;R&S[;R&S[R,;R&S[S[S[,,R,;R&S[R,;R&S[S[,R,;R &S[;R &S[;R &#) rNrHrIrr<r=r>r?rUrr;)r r!r3rr@rrA)r"r#s"r$r%DiaConfig.__annotate__Us6%t+d2978%t+d299:;<#=>*#?@S /D(/AB*#CD9t#*EF#GHIr')rIrUrH)r)r*r+r,rCr-keys_to_ignore_at_inferencer r3 sub_configsrHrIrr<r=r>r?rUrr;rYrarer.r/r0 __classcell__)r\r#s@@r$rErEUsx(J#4"5#%57GIYZK59N59NH##L+/L#L&*M#I"(HT #_r'rEN)rEr r3)rChuggingface_hub.dataclassesrconfiguration_utilsrmodeling_rope_utilsrutilsrr get_loggerr)rVr r3rE__all__r(r'r$rrs.31,   H %/0$'$1$"/0#$'#$1#$L/0O# O#1O#d @r'