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

fix median index issue

Browse Source
This commit is contained in:
Rajat Sen
2024-11-09 20:28:02 +00:00
parent 0706b0fb5c
commit c0709fdd69
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/timesfm/timesfm_jax.py
+1
View File
@@ -89,6 +89,7 @@ class TimesFmJax(timesfm_base.TimesFmBase):
self._pmapped_decode = None self._pmapped_decode = None
self._model = None self._model = None
self._train_state = None self._train_state = None
self._median_index = -1
def load_from_checkpoint( def load_from_checkpoint(
self, self,