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

fix typo

Browse Source
This commit is contained in:
siriuz42
2025-10-07 21:26:57 +00:00
parent ff2cefaa26
commit 1436ad859c
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/timesfm/timesfm_2p5/timesfm_2p5_flax.py
+1 -1
View File
@@ -509,7 +509,7 @@ class TimesFM_2p5_200M_flax(timesfm_2p5_base.TimesFM_2p5):
self.model.o, self.model.o,
new_horizon := math.ceil(fc.max_horizon / self.model.o) * self.model.o, new_horizon := math.ceil(fc.max_horizon / self.model.o) * self.model.o,
) )
forecast_config = dataclasses.replace(fc, max_horizon=new_horizon) fc = dataclasses.replace(fc, max_horizon=new_horizon)
if fc.max_context + fc.max_horizon > self.model.config.context_limit: if fc.max_context + fc.max_horizon > self.model.config.context_limit:
raise ValueError( raise ValueError(
"Context + horizon must be less than the context limit." "Context + horizon must be less than the context limit."
src/timesfm/timesfm_2p5/timesfm_2p5_torch.py
+1 -1
View File
@@ -356,7 +356,7 @@ class TimesFM_2p5_200M_torch(timesfm_2p5_base.TimesFM_2p5, ModelHubMixin):
self.model.o, self.model.o,
new_horizon := math.ceil(fc.max_horizon / self.model.o) * self.model.o, new_horizon := math.ceil(fc.max_horizon / self.model.o) * self.model.o,
) )
forecast_config = dataclasses.replace(fc, max_horizon=new_horizon) fc = dataclasses.replace(fc, max_horizon=new_horizon)
if fc.max_context + fc.max_horizon > self.model.config.context_limit: if fc.max_context + fc.max_horizon > self.model.config.context_limit:
raise ValueError( raise ValueError(
"Context + horizon must be less than the context limit." "Context + horizon must be less than the context limit."