wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-06 22:51:18 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Fix logic flaw identified by autonomous review

Browse Source
This commit is contained in:
fliptrigga13 2026-05-29 18:44:59 -04:00
parent ec1896aceb
commit f2b7df6bf0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/mmaudio/vae/vae_modules.py
View File

@ -17,7 +17,7 @@ def normalize(x, dim=None, eps=1e-4):
if dim is None: if dim is None:
dim = list(range(1, x.ndim)) dim = list(range(1, x.ndim))
norm = torch.linalg.vector_norm(x, dim=dim, keepdim=True, dtype=torch.float32) norm = torch.linalg.vector_norm(x, dim=dim, keepdim=True, dtype=torch.float32)
norm = torch.add(eps, norm, alpha=math.sqrt(norm.numel() / x.numel())) norm = torch.add(norm, eps) / math.sqrt(norm.numel() / x.numel())
return x / norm.to(x.dtype) return x / norm.to(x.dtype)
class ResnetBlock1D(nn.Module): class ResnetBlock1D(nn.Module):
@ -118,4 +118,4 @@ class Downsample1D(nn.Module):
if self.with_conv: if self.with_conv:
x = self.conv2(x) x = self.conv2(x)
return x return x