fix binary wheel installation when nvcc is not available (#448)

aab603af · Aman Gupta Karmani · GitHub · f8dccfc9 · aab603af
Unverified Commit aab603af authored Aug 14, 2023 by Aman Gupta Karmani Committed by GitHub Aug 14, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 2 deletions

setup.py setup.py +0 -2

No files found.
--- a/setup.py
+++ b/setup.py
@@ -247,8 +247,6 @@ class CachedWheelsCommand(_bdist_wheel):
        if FORCE_BUILD:
            return super().run()

-        raise_if_cuda_home_none("flash_attn")
-
        # Determine the version numbers that will be used to determine the correct wheel
        # We're using the CUDA version used to build torch, not the one currently installed
        # _, cuda_version_raw = get_cuda_bare_metal_version(CUDA_HOME)