CI: Pin flash-attn-4 for diffusion CI consistency (#28838)
This commit is contained in:
@@ -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
|
"easydict", # Required by remote model code (e.g. DeepSeek-OCR) loaded via trust_remote_code; validated by transformers 5.4+ check_imports
|
||||||
"einops",
|
"einops",
|
||||||
"fastapi",
|
"fastapi",
|
||||||
"flash-attn-4>=4.0.0b9",
|
"flash-attn-4==4.0.0b15",
|
||||||
"flashinfer_cubin==0.6.12",
|
"flashinfer_cubin==0.6.12",
|
||||||
"flashinfer_python[cu13]==0.6.12", # keep it aligned with jit-cache version in Dockerfile
|
"flashinfer_python[cu13]==0.6.12", # keep it aligned with jit-cache version in Dockerfile
|
||||||
"gguf",
|
"gguf",
|
||||||
|
|||||||
Reference in New Issue
Block a user