Skip to content

Commit 54b518c

Browse files
committed
tmp: force torch<2.11.0 in cu12 to run on GA
1 parent 61a7889 commit 54b518c

2 files changed

Lines changed: 4369 additions & 3875 deletions

File tree

pyproject.toml

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,9 +52,9 @@ gpu-cu12 = ["cupy-cuda12x"]
5252
gpu-cu13 = ["cupy-cuda13x"]
5353
stat = ["pytensor", "pymc"]
5454
deep = ["torch", "jax"]
55-
deep-cu126 = ["torch", "jax[cuda12]"]
56-
deep-cu128 = ["torch", "jax[cuda12]"]
57-
deep-cu13 = ["torch>=2.11", "jax[cuda13]"]
55+
deep-cu126 = ["torch<2.11.0", "jax[cuda12]"]
56+
deep-cu128 = ["torch<2.11.0", "jax[cuda12]"]
57+
deep-cu13 = ["torch>=2.11.0", "jax[cuda13]"]
5858

5959
[dependency-groups]
6060
dev = [
@@ -101,6 +101,7 @@ conflicts = [
101101
{ extra = "deep" },
102102
{ extra = "deep-cu126" },
103103
{ extra = "deep-cu128" },
104+
{ extra = "deep-cu13" },
104105
],
105106
]
106107

0 commit comments

Comments
 (0)