squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Aman Gupta Karmani 2023-08-14 14:54:26 -07:00 committed by GitHub
parent f8dccfc90a
commit aab603af4f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -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)