From 90370e3ff2ddc6842d2b20cb2a5327726907852b Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Mon, 28 Oct 2024 13:33:10 -0700 Subject: [PATCH] Update torch-ver in the sd-tests --- .github/workflows/nv-sd.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/nv-sd.yml b/.github/workflows/nv-sd.yml index b6d208aed5cb..9849954ca271 100644 --- a/.github/workflows/nv-sd.yml +++ b/.github/workflows/nv-sd.yml @@ -64,7 +64,7 @@ jobs: run: | unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch cd tests - python -m pytest --color=yes --durations=0 --verbose -rF -m 'stable_diffusion' -k "TestStableDiffusion" unit/ --torch_ver="2.0" --cuda_ver="12" + python -m pytest --color=yes --durations=0 --verbose -rF -m 'stable_diffusion' -k "TestStableDiffusion" unit/ --torch_ver="2.1" --cuda_ver="12" - name: Open GitHub issue if weekly CI fails if: ${{ failure() && (github.event_name == 'schedule') }}