From 501ddd5614d78fe23dc0eab4c03200f99c43d41b Mon Sep 17 00:00:00 2001 From: Rajat Sen Date: Wed, 1 Oct 2025 18:10:10 +0000 Subject: [PATCH] adding these changes to default config. --- src/timesfm/timesfm_2p5/timesfm_2p5_base.py | 50 --------------------- 1 file changed, 50 deletions(-) diff --git a/src/timesfm/timesfm_2p5/timesfm_2p5_base.py b/src/timesfm/timesfm_2p5/timesfm_2p5_base.py index 1ddb5b0..f438e40 100644 --- a/src/timesfm/timesfm_2p5/timesfm_2p5_base.py +++ b/src/timesfm/timesfm_2p5/timesfm_2p5_base.py @@ -82,56 +82,6 @@ def linear_interpolation(arr): class TimesFM_2p5_200M_Definition: """Framework-agnostic config of TimesFM 2.5.""" - context_limit = 16384 - input_patch_len: int = 32 - output_patch_len: int = 128 - output_quantile_len: int = 1024 - quantiles: list[float] = dataclasses.field( - default_factory=lambda: [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9] - ) - decode_index: int = 5 - tokenizer: ResidualBlockConfig = ResidualBlockConfig( - input_dims=64, - hidden_dims=1280, - output_dims=1280, - use_bias=True, - activation="swish", - ) - stacked_transformers: StackedTransformersConfig = StackedTransformersConfig( - num_layers=20, - transformer=TransformerConfig( - model_dims=1280, - hidden_dims=1280, - num_heads=16, - attention_norm="rms", - feedforward_norm="rms", - qk_norm="rms", - use_bias=False, - use_rotary_position_embeddings=True, - ff_activation="swish", - fuse_qkv=False, - ), - ) - output_projection_point: ResidualBlockConfig = ResidualBlockConfig( - input_dims=1280, - hidden_dims=1280, - output_dims=1280, - use_bias=False, - activation="swish", - ) - output_projection_quantiles: ResidualBlockConfig = ResidualBlockConfig( - input_dims=1280, - hidden_dims=1280, - output_dims=10240, - use_bias=False, - activation="swish", - ) - - -@dataclasses.dataclass(frozen=True) -class TimesFM_2p5_new_200M_Definition: - """Framework-agnostic config of TimesFM 2.5.""" - context_limit = 16384 input_patch_len: int = 32 output_patch_len: int = 128