
    Z j	                     p    S r 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)zNemotron model configuration    )strict   )PreTrainedConfig)RopeParameters)auto_docstringzthhaus/nemotron3-8b)
checkpointc                     ^  \ rS rSr% SrSrS/r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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r\\-  S-  \	S#'   S!r \\	S$'   S%r!\\-  \	S&'   S!r"\\	S''   U 4S( jr#S)r$U =r%$ )*NemotronConfig   aq  
Example:

```python
>>> from transformers import NemotronModel, NemotronConfig

>>> # Initializing a Nemotron nemotron-15b style configuration
>>> configuration = NemotronConfig()

>>> # Initializing a model from the nemotron-15b style configuration
>>> model = NemotronModel(configuration)

>>> # Accessing the model configuration
>>> configuration = model.config
```nemotronpast_key_valuesi  
vocab_sizei   hidden_sizei `  intermediate_size    num_hidden_layers0   num_attention_headsNhead_dimnum_key_value_headsrelu2
hidden_acti   max_position_embeddingsgS!uq?initializer_rangegh㈵>norm_epsT	use_cachepad_token_id   bos_token_idr   eos_token_idFtie_word_embeddingsrope_parametersattention_biasg        attention_dropoutmlp_biasc                    > U R                   b  U R                   OU R                  U R                  -  U l         UR                  SS5        [        TU ]  " S0 UD6  g )Npartial_rotary_factorg      ? )r   r   r   
setdefaultsuper__post_init__)selfkwargs	__class__s     ڄ/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/models/nemotron/configuration_nemotron.pyr+   NemotronConfig.__post_init__C   sL    )-)BHXHX\`\t\tHt137''    )r   )&__name__
__module____qualname____firstlineno____doc__
model_typekeys_to_ignore_at_inferencer   int__annotations__r   r   r   r   r   r   r   strr   r   floatr   r   boolr   r   r    listr!   r"   r   dictr#   r$   r%   r+   __static_attributes____classcell__)r.   s   @r/   r
   r
      s     J#4"5JK"s"s!!HcDj&*t*J#'S'%u%HeIt#L#*# L#* +,L#S	/D(, %%48O^d*T18 ND %(us{(Hd( (r1   r
   N)r6   huggingface_hub.dataclassesr   configuration_utilsr   modeling_rope_utilsr   utilsr   r
   __all__r(   r1   r/   <module>rG      sK    # . 3 1 # 01,(% ,(  2,(^ 
r1   