From a7facedef9007a04d35fc1dbcdce7f232fb52fb5 Mon Sep 17 00:00:00 2001
From: AbdBarho <ka70911@gmail.com>
Date: Fri, 14 Oct 2022 16:14:50 +0200
Subject: [PATCH] h

---
 .github/workflows/docker.yml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index 4db5a76..ddbf709 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -13,7 +13,9 @@ 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: "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.1"
+        NVCC_FLAGS: "--use_fast_math -DXFORMERS_MEM_EFF_ATTENTION_DISABLE_BACKWARD"
     steps:
       - run: |
           apt-get update