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

Merge pull request #18 from gahdritz/main

Browse Source 
Slightly improve installation process
This commit is contained in:
Tri Dao 2022-07-02 23:33:14 -07:00 committed by GitHub
commit 01947bc93b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
flash_attn/__init__.py Normal file
View File

6
setup.py
View File

@ -168,5 +168,9 @@ setup(
], ],
ext_modules=ext_modules, ext_modules=ext_modules,
cmdclass={"build_ext": BuildExtension} if ext_modules else {}, cmdclass={"build_ext": BuildExtension} if ext_modules else {},
python_requires=">=3.7" python_requires=">=3.7",
install_requires=[
"torch",
"einops",
],
) )