CI: Pin flash-attn-4 for diffusion CI consistency (#28838)

2026-06-21 21:13:41 +08:00
parent 3975ea5ac7
commit a51d56d948
1 changed files with 1 additions and 1 deletions
@@ -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",