Merge pull request #18 from gahdritz/main

Slightly improve installation process
2022-07-02 23:33:14 -07:00 · 2022-07-02 23:33:14 -07:00 · 01947bc93b
commit 01947bc93b
parent f66603cb6f af4a9ce024
2 changed files with 5 additions and 1 deletions
--- 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",
+    ],
 )