fix median index issue

2024-11-09 20:28:02 +00:00
parent 0706b0fb5c
commit c0709fdd69
1 changed files with 1 additions and 0 deletions
@@ -89,6 +89,7 @@ class TimesFmJax(timesfm_base.TimesFmBase):
    self._pmapped_decode = None
    self._model = None
    self._train_state = None
+    self._median_index = -1

  def load_from_checkpoint(
      self,