From f2a975d5e363ab49442cc39e0b044a365b34e1d3 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Wed, 10 Jan 2024 01:17:20 -0500
Subject: [PATCH] forgot flash-attention in check

---
 setup.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 04a51550d..235018dcc 100644
--- a/setup.py
+++ b/setup.py
@@ -12,7 +12,10 @@ def parse_requirements():
         lines = [r.strip() for r in requirements_file.readlines()]
         for line in lines:
             is_extras = (
-                "flash-attn" in line or "deepspeed" in line or "mamba-ssm" in line
+                "flash-attn" in line
+                or "flash-attention" in line
+                or "deepspeed" in line
+                or "mamba-ssm" in line
             )
             if line.startswith("--extra-index-url"):
                 # Handle custom index URLs