diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py index db46bfb..6763b5f 100644 --- a/wanvideo/modules/model.py +++ b/wanvideo/modules/model.py @@ -407,7 +407,7 @@ class WanSelfAttention(nn.Module): self.v = nn.Linear(in_features, out_features) self.o = nn.Linear(in_features, out_features) - if hasattr(torch.nn, 'RMSNorm'): + if False: # disable for now, seems to degrade quality self.norm_q = WanFusedRMSNorm(out_features, eps=eps) if qk_norm else nn.Identity() self.norm_k = WanFusedRMSNorm(out_features, eps=eps) if qk_norm else nn.Identity() else: