Skip to content

Commit 8c1d175

Browse files
authored
pyproject: use torch>=2.7 (#175)
1 parent 63c78fe commit 8c1d175

File tree

3 files changed

+3
-6
lines changed

3 files changed

+3
-6
lines changed

.github/workflows/lint.yaml

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,6 @@ jobs:
2323
2424
sudo apt-get install -y protobuf-compiler
2525
26-
# use RC build
27-
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cu128
28-
2926
pip install lintrunner lintrunner-adapters
3027
lintrunner init
3128

.github/workflows/unittest.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,11 @@ jobs:
1515
- runs-on: "linux.2xlarge"
1616
gpu-arch-type: "cpu"
1717
gpu-arch-version: ""
18-
torch-version: "test"
18+
torch-version: "stable"
1919
- runs-on: "linux.g5.12xlarge.nvidia.gpu"
2020
gpu-arch-type: "cuda"
2121
gpu-arch-version: "12.4"
22-
torch-version: "test"
22+
torch-version: "stable"
2323
- runs-on: "linux.g5.12xlarge.nvidia.gpu"
2424
gpu-arch-type: "cuda"
2525
gpu-arch-version: "12.4"

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ classifiers = [
1212
]
1313
dynamic = ["version"]
1414
dependencies = [
15-
"torch"
15+
"torch>=2.7"
1616
]
1717

1818
[project.urls]

0 commit comments

Comments
 (0)