From de6b95c62578ff894a06ab9f9c5d070dbf7ab064 Mon Sep 17 00:00:00 2001 From: Jesse Cai Date: Wed, 24 Jul 2024 11:00:01 -0700 Subject: [PATCH] update --- .github/workflows/regression_test.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/regression_test.yml b/.github/workflows/regression_test.yml index 1698fb424f..119d228085 100644 --- a/.github/workflows/regression_test.yml +++ b/.github/workflows/regression_test.yml @@ -23,7 +23,7 @@ jobs: include: - name: CUDA 2.2.2 runs-on: linux.g5.12xlarge.nvidia.gpu - torch-spec: 'torch==2.2.2 numpy==1.26' + torch-spec: 'torch==2.2.2 "numpy<2" ' gpu-arch-type: "cuda" gpu-arch-version: "12.1" - name: CUDA 2.3 @@ -38,7 +38,7 @@ jobs: gpu-arch-version: "12.1" - name: CPU 2.2.2 runs-on: linux.4xlarge - torch-spec: 'torch==2.2.2 --index-url https://download.pytorch.org/whl/cpu numpy==1.26' + torch-spec: 'torch==2.2.2 --index-url https://download.pytorch.org/whl/cpu "numpy<2" ' gpu-arch-type: "cpu" gpu-arch-version: "" - name: CPU 2.3