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

Clean setup.py imports

Browse Source
This commit is contained in:
Pierce Freeman 2023-06-07 17:26:13 -07:00
parent eb812c205b
commit 9af165c389
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@ -9,9 +9,7 @@ from packaging.version import parse, Version
import platform
from setuptools import setup, find_packages
from setuptools.command.build import build
import subprocess
from setuptools.command.bdist_egg import bdist_egg
import urllib.request
import urllib.error
@ -214,7 +212,7 @@ class CachedWheelsCommand(_bdist_wheel):
"""
def run(self):
if FORCE_BUILD:
return build.run(self)
return super().run()
raise_if_cuda_home_none("flash_attn")