[diffusion] refactor: refactor fuse qkv with QKVParallelLinear linear (#15090)

Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
Xiaoyu Zhang
2025-12-15 00:33:29 +08:00
committed by GitHub
co-authored by Mick
parent d277a86dea
commit 64b5c3ab90
9 changed files with 118 additions and 285 deletions
@@ -692,10 +692,6 @@ class TransformerLoader(ComponentLoader):
model = model.eval()
if hasattr(model, "fuse_qkv_projections"):
logger.info("Fusing QKV projections for better performance")
model.fuse_qkv_projections()
return model