freedak/timesfm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adding these changes to default config.

Browse Source
This commit is contained in:
Rajat Sen
2025-10-01 18:08:11 +00:00
parent c58e6b6bca
commit 08918788ad
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/timesfm/timesfm_2p5/timesfm_2p5_torch.py
+1 -7
View File
@@ -262,16 +262,10 @@ class TimesFM_2p5_200M_torch_module(nn.Module):
return outputs 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): class TimesFM_2p5_200M_torch(timesfm_2p5_base.TimesFM_2p5, ModelHubMixin):
"""PyTorch implementation of TimesFM 2.5 with 200M parameters.""" """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 @classmethod
def _from_pretrained( def _from_pretrained(