From a6609df1d50b7306837fa32c878d08c4423f79fd Mon Sep 17 00:00:00 2001 From: Yichen Zhou Date: Wed, 8 Oct 2025 16:03:18 -0700 Subject: [PATCH] Update timesfm_2p5_flax.py bug fix --- src/timesfm/timesfm_2p5/timesfm_2p5_flax.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/timesfm/timesfm_2p5/timesfm_2p5_flax.py b/src/timesfm/timesfm_2p5/timesfm_2p5_flax.py index 51d6a3c..8da19b8 100644 --- a/src/timesfm/timesfm_2p5/timesfm_2p5_flax.py +++ b/src/timesfm/timesfm_2p5/timesfm_2p5_flax.py @@ -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,