diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index 0650d02..515914d 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -52,7 +52,7 @@ jobs:
         # Without this we get import error (undefined symbol: _ZN3c105ErrorC2ENS_14SourceLocationESs)
         # when building without C++11 ABI and using it on nvcr images.
         cxx11_abi: ["FALSE", "TRUE"]
-        arch: ["80", "90", "100", "120"]
+        arch: ["80", "90", "100"]
         include:
             - torch-version: "2.9.0.dev20250904"
               cuda-version: "13.0"
diff --git a/setup.py b/setup.py
index 95015cd..6bda267 100644
--- a/setup.py
+++ b/setup.py
@@ -79,7 +79,7 @@ def should_skip_cuda_build():
 
 @functools.lru_cache(maxsize=None)
 def cuda_archs():
-    return os.getenv("FLASH_DMATTN_CUDA_ARCHS", "80;90;100;120").split(";")
+    return os.getenv("FLASH_DMATTN_CUDA_ARCHS", "80;90;100").split(";")
 
 
 def detect_preferred_sm_arch() -> Optional[str]: