^jdZddlmZddlmZddlmZddlmZddl m Z ed eGd d eZ ed eGd deZ dd gZ y)zMusicGen model configuration)ClassVar)strict)PreTrainedConfig)auto_docstring) AutoConfigzfacebook/musicgen-small) checkpointceZdZUdZdZdgZdZeed<dZ eed<dZ eed<d Z eed <d Z eed <d Z eezed<dZeed<dZeed<dZeed<dZeezed<d Zeezed<d Zeezed<dZeed<dZeed<dZeed<dZeed <dZed!zed"<dZed!zed#<d!Zeeezd!zed$<dZeed%<dZ eed&<dZ!eed'<d!Z"ed!zed(<d)Z#y!)*MusicgenDecoderConfigmusicgen_decoderdecoder_configpast_key_valuesi vocab_sizemax_position_embeddingsnum_hidden_layersiffn_dimnum_attention_headsg layerdropT use_cachegeluactivation_functioni hidden_sizeg?dropoutattention_dropoutactivation_dropout{Gz?initializer_factorFscale_embedding num_codebooksaudio_channelsN pad_token_id bos_token_id eos_token_idtie_word_embeddings is_decoderadd_cross_attentioncross_attention_hidden_sizecR|jdvrtd|jdy)zOPart of `@strict`-powered validation. Validates the architecture of the config.)r$rz4Expected 1 (mono) or 2 (stereo) audio channels, got z channels.N)r% ValueErrorselfs ~/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/transformers/models/musicgen/configuration_musicgen.pyvalidate_architecturez+MusicgenDecoderConfig.validate_architecture8s3   f ,STXTgTgShhrst t -)$__name__ __module__ __qualname__ model_typebase_config_keykeys_to_ignore_at_inferencerint__annotations__rrrrrfloatrboolrstrrrrrr r!r#r%r&r'r(listr)r*r+r,r2r3r1r r s6$J&O#4"5J#'S'sGS!! Ius{ It%%KGUS[%(us{(&) ) $$!OT!M3NC#L#*##L#*#+/L#S /D(/ %%J %%.2t2ur3r ceZdZUdZdZeeed<eee dZ ee ee e fed<dZeeed<dZe e zed <dZe e zed <dZe e zed <d Zeed <fdZedZxZS)MusicgenConfiga, text_encoder (`Union[dict, `PretrainedConfig`]`): An instance of a configuration object that defines the text encoder config. audio_encoder (`Union[dict, `PretrainedConfig`]`): An instance of a configuration object that defines the audio encoder config. decoder (`Union[dict, `PretrainedConfig`]`): An instance of a configuration object that defines the decoder config. Example: ```python >>> from transformers import ( ... MusicgenConfig, ... MusicgenDecoderConfig, ... T5Config, ... EncodecConfig, ... MusicgenForConditionalGeneration, ... ) >>> # Initializing text encoder, audio encoder, and decoder model configurations >>> text_encoder_config = T5Config() >>> audio_encoder_config = EncodecConfig() >>> decoder_config = MusicgenDecoderConfig() >>> configuration = MusicgenConfig( ... text_encoder=text_encoder_config, ... audio_encoder=audio_encoder_config, ... decoder=decoder_config, ... ) >>> # Initializing a MusicgenForConditionalGeneration (with random weights) from the facebook/musicgen-small style configuration >>> model = MusicgenForConditionalGeneration(configuration) >>> # Accessing the model configuration >>> configuration = model.config >>> config_text_encoder = model.config.text_encoder >>> config_audio_encoder = model.config.audio_encoder >>> config_decoder = model.config.decoder >>> # Saving the model, including its configuration >>> model.save_pretrained("musicgen-model") >>> # loading model and config from pretrained folder >>> musicgen_config = MusicgenConfig.from_pretrained("musicgen-model") >>> model = MusicgenForConditionalGeneration.from_pretrained("musicgen-model", config=musicgen_config) ```musicgenr7) text_encoder audio_encoderdecoder sub_configsThas_no_defaults_at_initNrDrErFrr c t|jtrA|jjd}t j |fi|j|_n%|jt d|jdt|jtrA|jjd}t j |fi|j|_n%|jt d|jdt|jtrtdi|j|_ n|jt|_ d|_ t|4di|y)Nr7zA configuration of type z: cannot be instantiated because text_encoder is not passedz; cannot be instantiated because audio_encoder is not passedTr@) isinstancerDdictpopr for_modelr.r7rErFr is_encoder_decodersuper __post_init__)r0kwargstext_encoder_model_typeaudio_encoder_model_type __class__s r1rPzMusicgenConfig.__post_init__}s8 d'' .&*&7&7&;&;L&I # * 4 45L bPTPaPa bD     &*4??*;;uv  d(($ /'+'9'9'='=l'K $!+!5!56N!eRVRdRd!eD     '*4??*;;vw  dllD )0@4<<@DL \\ !02DL"& ''r3c.|jjS)N)rE sampling_rater/s r1rVzMusicgenConfig.sampling_rates!!///r3)r4r5r6__doc__r7rr>r;r r rGrKtyperrHr=rDrErFr r<rPpropertyrV __classcell__)rTs@r1rBrB>s- ^!+J *"#(@K$sD)9$::;< /3Xd^2,0L$))0-1M4**1'+GT$ $+ $$(200r3rBN)rWtypingrhuggingface_hub.dataclassesrconfiguration_utilsrutilsrauto.configuration_autor r rB__all__r@r3r1ras#.3#045 u, u6 uF45Y0%Y06Y0x 4 5r3