From cc06016cae712767032eaa19035d040aaed9dfeb Mon Sep 17 00:00:00 2001 From: jainapurva Date: Thu, 23 Jan 2025 15:58:05 -0800 Subject: [PATCH] Fixed nightly version --- .github/workflows/float8_test.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/float8_test.yml b/.github/workflows/float8_test.yml index 52a1b478e8..42a5861624 100644 --- a/.github/workflows/float8_test.yml +++ b/.github/workflows/float8_test.yml @@ -30,7 +30,7 @@ jobs: gpu-arch-version: "12.4" - name: H100 runs-on: linux.aws.h100 - torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu124' + torch-spec: '--pre torch==2.6.0 --index-url https://download.pytorch.org/whl/nightly/cu124' gpu-arch-type: "cuda" gpu-arch-version: "12.4"