
    i                     v    d dl mZ ddlmZ ddlmZ ddlmZmZ  ed      e G d	 d
e                    Z	d
gZ
y)    )strict   )PreTrainedConfig)auto_docstring   )CONFIG_MAPPING
AutoConfigznvidia/music-flamingo-2601-hf)
checkpointc                        e Zd ZU dZdZeedZdZee	z  dz  e
d<   dZee	z  dz  e
d<   dZee
d<   d	Zee
d
<   dZee
d<   dZee
d<   dZee
d<   dZee
d<   dZedz  e
d<    fdZ xZS )MusicFlamingoConfiga  
    audio_bos_token_id (`int`, *optional*, defaults to 151670):
        The beginning-of-audio token index used to mark the start of audio spans.
    audio_eos_token_id (`int`, *optional*, defaults to 151671):
        The end-of-audio token index used to mark the end of audio spans.
    audio_frame_step (`float`, *optional*, defaults to 0.01):
        Duration in seconds of one input mel frame (trained with hop_length 160 at sampling_rate 16000).

    Example:

    ```python
    >>> from transformers import MusicFlamingoForConditionalGeneration, MusicFlamingoConfig, AudioFlamingo3EncoderConfig, Qwen2Config

    >>> # Initializing an MusicFlamingoEncoder config
    >>> audio_config = AudioFlamingo3EncoderConfig()

    >>> # Initializing a Qwen2 config
    >>> text_config = Qwen2Config()

    >>> # Initializing an MusicFlamingo configuration
    >>> configuration = MusicFlamingoConfig(audio_config, text_config)

    >>> # Initializing a model from the musicflamingo style configuration
    >>> model = MusicFlamingoForConditionalGeneration(configuration)

    >>> # Accessing the model configuration
    >>> configuration = model.config
    ```musicflamingo)audio_configtext_configNr   r   iuP audio_token_idgeluprojector_hidden_actTprojector_biasivP audio_bos_token_idiwP audio_eos_token_idg{Gz?audio_frame_steprope_parametersc                    t        | j                  t              rT| j                  j                  dd      | j                  d<   t	        | j                  d      d	i | j                  | _        n| j                  t	        d          | _        t        | j
                  t              rT| j
                  j                  dd      | j
                  d<   t	        | j
                  d      d	i | j
                  | _        n| j
                  t	        d          | _        t        |   d	i | | j                  dddd| _        | j                  d   | _	        | j                  j                  | _        y )
N
model_typemusicflamingo_encoderqwen2defaulti  g?)	rope_type
rope_thetapartial_rotary_factorr    )
isinstancer   dictgetr   r   super__post_init__r   max_position_embeddingshidden_sizehead_dim)selfkwargs	__class__s     /var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/transformers/models/musicflamingo/configuration_musicflamingo.pyr%   z!MusicFlamingoConfig.__post_init__K   sD   d''..2.?.?.C.CLRi.jDl+ .t/@/@/N O dRVRcRc dD& ./F G IDd&&--1-=-=-A-A,PW-XD\*-d.>.>|.LMaPTP`P`aD%-g68D'''1:$il#mD '+';';L'I$))55    )__name__
__module____qualname____doc__r   r	   sub_configsr   r"   r   __annotations__r   r   intr   strr   boolr   r   r   floatr   r%   __classcell__)r+   s   @r,   r   r      s    : !J#-jIK37L$))D0726K((4/6 NC  &#&ND$$$$"e"#'OTD['6 6r-   r   N)huggingface_hub.dataclassesr   configuration_utilsr   utilsr   autor   r	   r   __all__r    r-   r,   <module>r>      sH   . / 3 # - :;=6* =6  <=6@ !
!r-   