Skip to content

Commit 85c82e3

Browse files
committed
Update Natten for cuda to 0.20.1
1 parent 6900401 commit 85c82e3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ mlx = ["mlx"]
9494
install = "pip install --editable ."
9595
installnnja = "pip install git+https://github.com/brightbandtech/nnja-ai.git"
9696
installnat = "pip install natten"
97-
installnatcuda = "pip install natten==0.17.4+torch240cu124 -f https://shi-labs.com/natten/wheels/"
97+
installnatcuda = "pip install natten==0.20.1+torch270cu128 -f https://whl.natten.org"
9898
installpyg = "pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.7.0+cpu.html"
9999
installpygcuda = "pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.7.0+cuda128.html"
100100
test = "pytest"

0 commit comments

Comments
 (0)