From 08918788ade5cfa023db458999c8d68f9225c20a Mon Sep 17 00:00:00 2001 From: Rajat Sen Date: Wed, 1 Oct 2025 18:08:11 +0000 Subject: [PATCH] adding these changes to default config. --- src/timesfm/timesfm_2p5/timesfm_2p5_torch.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/src/timesfm/timesfm_2p5/timesfm_2p5_torch.py b/src/timesfm/timesfm_2p5/timesfm_2p5_torch.py index 7753850..897dbe7 100644 --- a/src/timesfm/timesfm_2p5/timesfm_2p5_torch.py +++ b/src/timesfm/timesfm_2p5/timesfm_2p5_torch.py @@ -262,16 +262,10 @@ class TimesFM_2p5_200M_torch_module(nn.Module): return outputs -class TimesFM_2p5_200M_torch_new_module(TimesFM_2p5_200M_torch_module): - """TimesFM 2.5 with 200M parameters.""" - - config = timesfm_2p5_base.TimesFM_2p5_new_200M_Definition() - - class TimesFM_2p5_200M_torch(timesfm_2p5_base.TimesFM_2p5, ModelHubMixin): """PyTorch implementation of TimesFM 2.5 with 200M parameters.""" - model: nn.Module = TimesFM_2p5_200M_torch_new_module() + model: nn.Module = TimesFM_2p5_200M_torch_module() @classmethod def _from_pretrained(