Skip to content

GitLab

Unverified Commit aab603af authored by Aman Gupta Karmani's avatar Aman Gupta Karmani Committed by GitHub
Browse files

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

parent f8dccfc9
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
setup.py
View file @ aab603af
...@@ -247,8 +247,6 @@ class CachedWheelsCommand(_bdist_wheel): ...@@ -247,8 +247,6 @@ class CachedWheelsCommand(_bdist_wheel):
if FORCE_BUILD: if FORCE_BUILD:
return super().run() return super().run()
raise_if_cuda_home_none("flash_attn")
# Determine the version numbers that will be used to determine the correct wheel # 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 # 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) # _, cuda_version_raw = get_cuda_bare_metal_version(CUDA_HOME)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment