apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

fix fa4 test (#6408)

Browse Source
This commit is contained in:
chen
2026-02-10 10:57:21 +08:00
committed by GitHub
parent 3ce842b55b
commit a8ffcaa068
3 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/model_executor/layers/attention/flash_attn_backend.py
+2 -2
View File
@@ -118,9 +118,9 @@ def flash_attn_func(
head_dim: int = 128,
version: Optional[int] = None,
):
if FLASH_ATTN_VERSION is None:
init_flash_attn_version()
if version is None:
if FLASH_ATTN_VERSION is None:
init_flash_attn_version()
version = FLASH_ATTN_VERSION
if version == 4:
assert (