From b522208e09102b250e298c8d14f317936d2076e0 Mon Sep 17 00:00:00 2001 From: doctorpangloss <2229300+doctorpangloss@users.noreply.github.com> Date: Wed, 22 Oct 2025 12:30:46 -0700 Subject: [PATCH] Fix linting issue --- comfy/ldm/mmaudio/vae/vae.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/comfy/ldm/mmaudio/vae/vae.py b/comfy/ldm/mmaudio/vae/vae.py index 65f625238..6723c24b1 100644 --- a/comfy/ldm/mmaudio/vae/vae.py +++ b/comfy/ldm/mmaudio/vae/vae.py @@ -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))