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

Update pyproject.toml

Browse Source 
Fix jax dependency issue
This commit is contained in:
Yichen Zhou
2024-07-12 13:42:46 -07:00
committed by GitHub
parent 5b03a34927
commit 0a40da3c52
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pyproject.toml
+1 -1
View File
@@ -5,7 +5,7 @@ name = "timesfm-jax"
description = "Open weights time-series foundation model from Google Research." description = "Open weights time-series foundation model from Google Research."
version = "1.0.1" version = "1.0.1"
dependencies = [ dependencies = [
"jax==0.4.26", "jax[cuda12]==0.4.26",
"paxml==1.4.0", "paxml==1.4.0",
"praxis==1.4.0", "praxis==1.4.0",
"jaxlib==0.4.26", "jaxlib==0.4.26",