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

Update timesfm_2p5_flax.py
2025-10-08 16:15:28 -07:00
parent ce168ffd1f a6609df1d5
commit 58e01ad82f
1 changed files with 1 additions and 1 deletions
@@ -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}."
      )

-    self.forecast_config = forecast_config
+    self.forecast_config = fc
    self.model.compile(
      context=self.forecast_config.max_context,
      horizon=self.forecast_config.max_horizon,