[BugFix][Diffusion] Scope Wan RMSNorm patch to NPU (#8047)

Signed-off-by: GUOGUO <xwdong1998@163.com>
This commit is contained in:
Dong1017 authored and GitHub committed 2026-09-23 15:08:02 +08:00
1 parent 75570ee3a6
commit 57d3e9fdf8
3 files changed
+16 -2

No files matched your search

@@ -535,7 +535,7 @@ def test_upsample_forward_only_fuses_nearest_2x() -> None:
def test_rms_norm_vae_substitute_is_not_matched() -> None:
from vllm_omni.diffusion.layers.norm import RMSNormVAE
from vllm_omni.diffusion.models.wan2_2.norm import RMSNormVAE
assert not fastpath_forwards.is_diffusers_rms_norm(RMSNormVAE(8, images=False))
_, vae = _build_pair(TINY_RESIDUAL, torch.float32)