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",