From a51d56d94892573b6973bf440c754e93967ea007 Mon Sep 17 00:00:00 2001
From: Mick <mickjagger19@icloud.com>
Date: Sun, 21 Jun 2026 21:13:41 +0800
Subject: [PATCH] CI: Pin flash-attn-4 for diffusion CI consistency (#28838)

---
 python/pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/pyproject.toml b/python/pyproject.toml
index 9be8cb833..0d4c89240 100755
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -30,7 +30,7 @@ dependencies = [
   "easydict",  # Required by remote model code (e.g. DeepSeek-OCR) loaded via trust_remote_code; validated by transformers 5.4+ check_imports
   "einops",
   "fastapi",
-  "flash-attn-4>=4.0.0b9",
+  "flash-attn-4==4.0.0b15",
   "flashinfer_cubin==0.6.12",
   "flashinfer_python[cu13]==0.6.12", # keep it aligned with jit-cache version in Dockerfile
   "gguf",