
    i
                     r    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
)    )strict   )PreTrainedConfig)RopeParameters)auto_docstringzkyutai/helium-1-preview)
checkpointc                      e Zd ZU dZdZdgZdZddddddddZdgd	gfd
dgd
gfd
gd
gf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
ed<   dZe
ed<   dZeed<   dZee
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z  d(z  ed)<   d*Ze
d(z  ed+<   d,Z e
e!e
   z  d(z  ed-<   d.Z"e
d(z  ed/<   d&Z#eed0<   d&Z$eed1<   y()2HeliumConfiga  
    Example:

    ```python
    >>> from transformers import HeliumModel, HeliumConfig
    >>> # Initializing a Helium 2b style configuration
    >>> configuration = HeliumConfig()
    >>> # Initializing a model from the Helium 2b style configuration
    >>> model = HeliumModel(configuration)
    >>> # Accessing the model configuration
    >>> configuration = model.config
    ```heliumpast_key_valuesg     j@colwiserowwise)zlayers.*.self_attn.q_projzlayers.*.self_attn.k_projzlayers.*.self_attn.v_projzlayers.*.self_attn.o_projzlayers.*.mlp.gate_projzlayers.*.mlp.up_projzlayers.*.mlp.down_proj	input_idsinputs_embedshidden_statesattention_mask)embed_tokenslayersnormi  
vocab_sizei 
  hidden_sizei  intermediate_size   num_hidden_layers   num_attention_headsnum_key_value_heads   head_dimsilu
hidden_actg        attention_dropouti   max_position_embeddingsg{Gz?initializer_rangeg:0yE>rms_norm_epsT	use_cacheFtie_word_embeddingsNrope_parametersr   pad_token_id   eos_token_id   bos_token_idattention_biasmlp_bias)%__name__
__module____qualname____doc__
model_typekeys_to_ignore_at_inferencedefault_thetabase_model_tp_planbase_model_pp_planr   int__annotations__r   r   r   r   r   r   r!   strr"   floatr#   r$   r%   r&   boolr'   r(   r   dictr)   r+   listr-   r.   r/        /var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/transformers/models/helium/configuration_helium.pyr
   r
      sd    J#4"5M%.%.%.%."+ )"+ &(9:#%568IJ!"_$56 JK!s!s!!!!HcJ%(us{(#'S'#u#L%It %%48O^d*T18 L#* +,L#S	/D(, L#*  ND HdrA   r
   N)
huggingface_hub.dataclassesr   configuration_utilsr   modeling_rope_utilsr   utilsr   r
   __all__r@   rA   rB   <module>rH      sH   " / 3 1 # 453# 3  63l 
rA   