
    Z j
                     l    S SK Jr  SSKJr  SSKJr  SSKJr  \" SS9\ " S S	\5      5       5       rS	/r	g
)    )strict   )PreTrainedConfig)RopeParameters)auto_docstringzkyutai/helium-1-preview)
checkpointc                      \ rS rSr% SrSrS/rSrSSSSSSSS.rS	/S
/4SS/S/4S/S/4S.r	Sr
\\S'   Sr\\S'   Sr\\S'   Sr\\S'   Sr\\S'   Sr\\S'   Sr\\S'   Sr\\S'   Sr\\-  \S'   Sr\\S'   S r\\S!'   S"r\\S#'   S$r\\S%'   S&r\\S''   S(r\\-  S(-  \S)'   S*r \S(-  \S+'   S,r!\\"\   -  S(-  \S-'   S.r#\S(-  \S/'   S&r$\\S0'   S&r%\\S1'   S2r&g()3HeliumConfig   aW  
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____firstlineno____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/   __static_attributes__r0       ڀ/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/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rC   r
   N)
huggingface_hub.dataclassesr   configuration_utilsr   modeling_rope_utilsr   utilsr   r
   __all__r0   rC   rD   <module>rJ      sH   " / 3 1 # 453# 3  63l 
rC   