tj dZddlmZddlmZddlmZedeGdd eZd gZd S) zTimesFM model configuration)strict)PreTrainedConfig)auto_docstringzgoogle/timesfm-2.0-500m-pytorch) checkpointcDeZdZUdZdZgZdZdZee d<dZ ee d<dZ ee d <d Z ee d <d Z ee d <dZee d<dZee d<dZee d<dZee d<dZee d<dZee d<dZeeeedfze d<dZee d<dZeeze d<dZee d<d Zee d!<d"Zee d#<d$Zee d%<d&S)' TimesFmConfiga( 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_lengthicontext_lengthhorizon_lengthr freq_size2num_hidden_layersi hidden_sizeintermediate_sizePhead_dimnum_attention_headsgư> tolerance rms_norm_eps) g?g?g333333?g?g?g333333?gffffff?g?g?. quantilesg@bApad_valgattention_dropoutuse_positional_embeddingg{Gz?initializer_range min_timescalei' max_timescaleN)__name__ __module__ __qualname____doc__ model_typekeys_to_ignore_at_inferenceis_encoder_decoderr int__annotations__r rrrrrrrrfloatrrlisttuplerrrboolrr!r"o/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/transformers/models/timesfm/configuration_timesfm.pyr r sk6J"$L#NCNCIssK!s!!!Hc!!!!IuL%1^ItE{U5#:..^^^!GU!!!%(us{(((%*d***#u###M3M3r1r N) r&huggingface_hub.dataclassesrconfiguration_utilsrutilsrr __all__r0r1r2r7s"!......333333######<===1 1 1 1 1 $1 1 >=1 h  r1