Skip to content

Commit de6b95c

Browse files
committed
update
1 parent 356d1a4 commit de6b95c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/regression_test.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ jobs:
2323
include:
2424
- name: CUDA 2.2.2
2525
runs-on: linux.g5.12xlarge.nvidia.gpu
26-
torch-spec: 'torch==2.2.2 numpy==1.26'
26+
torch-spec: 'torch==2.2.2 "numpy<2" '
2727
gpu-arch-type: "cuda"
2828
gpu-arch-version: "12.1"
2929
- name: CUDA 2.3
@@ -38,7 +38,7 @@ jobs:
3838
gpu-arch-version: "12.1"
3939
- name: CPU 2.2.2
4040
runs-on: linux.4xlarge
41-
torch-spec: 'torch==2.2.2 --index-url https://download.pytorch.org/whl/cpu numpy==1.26'
41+
torch-spec: 'torch==2.2.2 --index-url https://download.pytorch.org/whl/cpu "numpy<2" '
4242
gpu-arch-type: "cpu"
4343
gpu-arch-version: ""
4444
- name: CPU 2.3

0 commit comments

Comments
 (0)