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

Merge pull request #326 from google-research/siriuz42-patch-1

Browse Source 
Update timesfm_2p5_flax.py
This commit is contained in:
Rajat Sen
2025-10-08 16:15:28 -07:00
committed by GitHub
parent ce168ffd1f a6609df1d5
commit 58e01ad82f
1 changed files with 1 additions and 1 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
@@ -523,7 +523,7 @@ class TimesFM_2p5_200M_flax(timesfm_2p5_base.TimesFM_2p5):
f"Continuous quantile head is not supported for horizons > {self.model.os}." f"Continuous quantile head is not supported for horizons > {self.model.os}."
) )
self.forecast_config = forecast_config self.forecast_config = fc
self.model.compile( self.model.compile(
context=self.forecast_config.max_context, context=self.forecast_config.max_context,
horizon=self.forecast_config.max_horizon, horizon=self.forecast_config.max_horizon,