
    i 	                     v    d Z ddlmZ ddlmZ ddlmZ ddlmZ  ed      e G d	 d
e                    Z	d
gZ
y)zBitNet model configuration    )strict   )PreTrainedConfig)RopeParameters)auto_docstringzmicrosoft/bitnet-b1.58-2B-4T)
checkpointc                   j    e Zd ZU dZdZdgZdZ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<   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<   dZedz  e	d<   dZeee   z  dz  e	d <   d!Zee	d"<   d!Zee	d#<   d$Zeez  dz  e	d%<   dZee z  dz  e	d&<    fd'Z! xZ"S )(BitNetConfigas  
    ```python
    >>> from transformers import BitNetModel, BitNetConfig

    >>> # Initializing a BitNet style configuration
    >>> configuration = BitNetConfig()

    >>> # Initializing a model from the BitNet style configuration
    >>> model = BitNetModel(configuration)

    >>> # Accessing the model configuration
    >>> configuration = model.config
    ```
    bitnetpast_key_valuesg    Ai  
vocab_sizei 
  hidden_sizei   intermediate_size   num_hidden_layers   num_attention_heads   Nnum_key_value_headsrelu2
hidden_acti   max_position_embeddingsg{Gz?initializer_rangegh㈵>rms_norm_epsT	use_cachepad_token_idi  bos_token_idi eos_token_idFtie_word_embeddingsattention_biasg        attention_dropoutrope_parametersc                 ^    | j                   | j                  | _         t        |   di | y )N )r   r   super__post_init__)selfkwargs	__class__s     /var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/transformers/models/bitnet/configuration_bitnet.pyr&   zBitNetConfig.__post_init__?   s-    ##+'+'?'?D$''    )#__name__
__module____qualname____doc__
model_typekeys_to_ignore_at_inferencedefault_thetar   int__annotations__r   r   r   r   r   r   strr   r   floatr   r   boolr   r   r   listr   r    r!   r"   r   dictr&   __classcell__)r)   s   @r*   r
   r
      s    J#4"5MJK!s!s!!&'t'J#'S'#u#L%It#L#*#%L#*%+1L#S	/D(1 %% ND ,/us{T)/48O^d*T18( (r+   r
   N)r/   huggingface_hub.dataclassesr   configuration_utilsr   modeling_rope_utilsr   utilsr   r
   __all__r$   r+   r*   <module>r@      sK    ! . 3 1 # 9:,(# ,(  ;,(^ 
r+   