diff --git a/.github/workflows/xformers.yml b/.github/workflows/xformers.yml index e3537a7..6997401 100644 --- a/.github/workflows/xformers.yml +++ b/.github/workflows/xformers.yml @@ -12,7 +12,7 @@ jobs: DEBIAN_FRONTEND: noninteractive XFORMERS_DISABLE_FLASH_ATTN: 1 FORCE_CUDA: 1 - TORCH_CUDA_ARCH_LIST: "3.7;5.0;5.2;6.0;6.1+PTX;7.0;7.5+PTX;8.0;8.6+PTX" + TORCH_CUDA_ARCH_LIST: "6.0;6.1;6.2;7.0;7.2;7.5;8.0;8.6" steps: - run: | apt-get update