
    Z j                     d    S 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TimesFM model configuration    )strict   )PreTrainedConfig)auto_docstringzgoogle/timesfm-2.0-500m-pytorch)
checkpointc                   >   \ rS rSr% SrSr/ r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4   -  \
S'   Sr\\
S'   Sr\\	-  \
S'   Sr\\
S '   S!r\\
S"'   S#r\	\
S$'   S%r\	\
S&'   S'r g())TimesFmConfig   a  
patch_length (`int`, *optional*, defaults to 32):
    The length of one patch in the input sequence.
context_length (`int`, *optional*, defaults to 512):
    The length of the input context.
horizon_length (`int`, *optional*, defaults to 128):
    The length of the prediction horizon.
freq_size (`int`, *optional*, defaults to 3):
    The number of frequency embeddings.
tolerance (`float`, *optional*, defaults to 1e-06):
    The tolerance for the quantile loss.
quantiles (`list[float]`, *optional*, defaults to `[0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9]`):
    The quantiles to predict.
pad_val (`float`, *optional*, defaults to 1123581321.0):
    The value used to pad the predictions.
attention_dropout (`float`, *optional*, defaults to 0.0):
    The dropout probability for the attention scores.
use_positional_embedding (`bool`, *optional*, defaults to `False`):
    Whether to add positional embeddings.
min_timescale (`int`, *optional*, defaults to 1):
    The start of the geometric positional index. Determines the periodicity of
    the added signal.
max_timescale (`int`, *optional*, defaults to 10000):
    The end of the geometric positional index. Determines the frequency of the
    added signal.
timesfmF    patch_lengthi   context_length   horizon_lengthr   	freq_size2   num_hidden_layersi   hidden_sizeintermediate_sizeP   head_dim   num_attention_headsgư>	tolerancerms_norm_eps)	g?g?g333333?g?g      ?g333333?gffffff?g?g?.	quantilesg  @bApad_valg        attention_dropoutuse_positional_embeddingg{Gz?initializer_range   min_timescalei'  max_timescale N)!__name__
__module____qualname____firstlineno____doc__
model_typekeys_to_ignore_at_inferenceis_encoder_decoderr   int__annotations__r   r   r   r   r   r   r   r   r   floatr   r   listtupler   r   r   boolr    r"   r#   __static_attributes__r$       ڂ/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/models/timesfm/configuration_timesfm.pyr	   r	      s    6 J"$L#NCNCIssK!s!Hc!!IuL%1^ItE{U5#:..^!GU!%(us{(%*d*#u#M3M3r4   r	   N)	r)   huggingface_hub.dataclassesr   configuration_utilsr   utilsr   r	   __all__r$   r4   r5   <module>r:      sH    " . 3 # <=1 $ 1   >1 h 
r4   