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

readme update and not pinning cuda versin for flax

Browse Source
This commit is contained in:
Rajat Sen
2025-10-08 15:50:21 +00:00
parent cae3104aa6
commit bf23fb4880
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pyproject.toml
+1 -1
View File
@@ -28,7 +28,7 @@ flax = [
"einshape",
"orbax-checkpoint",
"jaxtyping",
"jax[cuda12]"
"jax[cuda]"
]
[tool.ruff]