Update timesfm_2p5_flax.py

bug fix
2025-10-08 16:03:18 -07:00
parent ce168ffd1f
commit a6609df1d5
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,