From 06ae39200b60d3482af8bd8bfc1a823c1b24e1da Mon Sep 17 00:00:00 2001
From: Casper <casperbh.96@gmail.com>
Date: Thu, 7 Dec 2023 07:36:52 +0100
Subject: [PATCH] Pin flash-attn to 2.3.3 (#919)

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index e3ee54350..986160273 100644
--- a/setup.py
+++ b/setup.py
@@ -46,7 +46,7 @@ setup(
     dependency_links=dependency_links,
     extras_require={
         "flash-attn": [
-            "flash-attn>=2.3.0",
+            "flash-attn==2.3.3",
         ],
         "deepspeed": [
             "deepspeed",