From 4c1a16f73cfbd2836187193d0cfe76f12199502c Mon Sep 17 00:00:00 2001
From: Casper <casperbh.96@gmail.com>
Date: Wed, 6 Dec 2023 22:33:41 +0100
Subject: [PATCH] Pin flash-attn to 2.3.3

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

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