Merge pull request #18 from gahdritz/main

Slightly improve installation process

Merge pull request #18 from gahdritz/main
Slightly improve installation process
01947bc9 · Tri Dao · GitHub · f66603cb · af4a9ce0 · 01947bc9
Unverified Commit 01947bc9 authored Jul 02, 2022 by Tri Dao Committed by GitHub Jul 02, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

flash_attn/__init__.py flash_attn/__init__.py +0 -0

setup.py setup.py +5 -1

No files found.
--- a/flash_attn/__init__.py
+++ b/flash_attn/__init__.py
--- a/setup.py
+++ b/setup.py
@@ -168,5 +168,9 @@ setup(
    ],
    ext_modules=ext_modules,
    cmdclass={"build_ext": BuildExtension} if ext_modules else {},
-    python_requires=">=3.7"
+    python_requires=">=3.7",
+    install_requires=[
+        "torch",
+        "einops",
+    ],
 )