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

Merge pull request #182 from google-research/rajat_dev

Browse Source 
Fix median index issue for jax
This commit is contained in:
Rajat Sen
2024-11-09 12:29:19 -08:00
committed by GitHub
parent 0706b0fb5c eb02472b88
commit 02bc2f2212
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/main.yml
+1 -1
View File
@@ -23,7 +23,7 @@ jobs:
# e.g. poetry version 0.1.${{ github.run_number }} # e.g. poetry version 0.1.${{ github.run_number }}
- name: Set Version number - name: Set Version number
run: | run: |
poetry version 1.2.2 poetry version 1.2.3
- name: Build and Publish to PyPI - name: Build and Publish to PyPI
run: | run: |
poetry config pypi-token.pypi ${{ secrets.PYPI_API_TOKEN }} poetry config pypi-token.pypi ${{ secrets.PYPI_API_TOKEN }}
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,