wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-10 14:20:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Fix linting issue

Browse Source
This commit is contained in:
doctorpangloss 2025-10-22 12:30:46 -07:00
parent aed4663bca
commit b522208e09
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/ldm/mmaudio/vae/vae.py
View File

@ -74,12 +74,17 @@ class VAE(nn.Module):
):
super().__init__()
data_mean = None
data_std = None
if data_dim == 80:
data_mean = torch.tensor(DATA_MEAN_80D, dtype=torch.float32)
data_std = torch.tensor(DATA_STD_80D, dtype=torch.float32)
elif data_dim == 128:
data_mean = torch.tensor(DATA_MEAN_128D, dtype=torch.float32)
data_std = torch.tensor(DATA_STD_128D, dtype=torch.float32)
else:
data_mean = torch.ones((1,))
data_std = torch.ones((1,))
self.register_buffer('data_mean', data_mean.view(1, -1, 1))
self.register_buffer('data_std', data_std.view(1, -1, 1))