diff --git a/pyproject.toml b/pyproject.toml
index 4e3321c..11f416a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -65,7 +65,7 @@ dev = [
 flash_attn = [
   # it's easier to install flash-attn from wheel rather than like this as extra
   # "https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.6/flash_attn-2.5.6+cu118torch2.0cxx11abiFALSE-cp311-cp311-linux_x86_64.whl",
-  "flash-attn==2.5.9.post1",
+  "flash-attn==2.6.1",
   "packaging",  # FIXME: temporary, until https://github.com/Dao-AILab/flash-attention/pull/937 is released
   "ninja"
 ]