From 440e9c49f261a29ae616f4a1dcf5d46aab346430 Mon Sep 17 00:00:00 2001 From: Gustaf Date: Sun, 3 Jul 2022 02:04:24 -0400 Subject: [PATCH 1/2] Add einops installation to setup.py --- setup.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 68baf9c..eabcf06 100644 --- 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", + ], ) From af4a9ce02425a2e9e99594d7b322bbf49a24bfbc Mon Sep 17 00:00:00 2001 From: Gustaf Date: Sun, 3 Jul 2022 02:04:55 -0400 Subject: [PATCH 2/2] Add missing __init__.py --- flash_attn/__init__.py | 0 1 file changed, 0 insertions(+), 0 deletions(-) create mode 100644 flash_attn/__init__.py diff --git a/flash_attn/__init__.py b/flash_attn/__init__.py new file mode 100644 index 0000000..e69de29