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

Merge pull request #310 from ram-from-tvl/fix/remove-lingvo

Browse Source 
Remove lingvo from v1 pax extra to avoid install failures on Colab
This commit is contained in:
Yichen Zhou
2025-09-25 13:06:00 -07:00
committed by GitHub
parent b810bbdf9f 098ad05360
commit 2e6db886fa
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
v1/pyproject.toml
+6 -4
View File
@@ -42,16 +42,18 @@ absl-py = ">=1.4.0"
safetensors = "^0.5.3"
[tool.poetry.extras]
pax = ["paxml", "lingvo", "jax", "jaxlib"]
# Note: `lingvo` is an optional Google-internal dependency with strict Python
# version and packaging constraints that cause install failures on some
# environments (Colab etc.). We omit it from the pax extra here so users can
# opt-in explicitly if they need it and have a compatible environment.
pax = ["paxml", "jax", "jaxlib"]
torch = ["torch"]
[tool.poetry.dependencies.paxml]
version = ">=1.4.0"
python = ">=3.10,<3.11"
[tool.poetry.dependencies.lingvo]
version = ">=0.12.7"
python = ">=3.10,<3.11"
[tool.poetry.dependencies.jax]
version = ">=0.4.26"