From aab603af4f65cd392fc2841067c27dcc78529844 Mon Sep 17 00:00:00 2001 From: Aman Gupta Karmani Date: Mon, 14 Aug 2023 14:54:26 -0700 Subject: [PATCH] fix binary wheel installation when nvcc is not available (#448) --- setup.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/setup.py b/setup.py index 6ee03f0..bf763ae 100644 --- 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)