From b6d64226cfdc470088f179de553ca8bbb3be7e1a Mon Sep 17 00:00:00 2001 From: AbdBarho Date: Fri, 14 Oct 2022 17:37:59 +0200 Subject: [PATCH] v --- .github/workflows/xformers.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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