From 674b69c291d247a1bb63a1e4b36b16fdacba86c2 Mon Sep 17 00:00:00 2001 From: doctorpangloss <2229300+doctorpangloss@users.noreply.github.com> Date: Wed, 22 Oct 2025 12:16:09 -0700 Subject: [PATCH] Fix linting errors, use register_buffer --- comfy/ldm/mmaudio/vae/vae.py | 12 ++++++------ comfy/ops.py | 3 ++- 2 files changed, 8 insertions(+), 7 deletions(-) diff --git a/comfy/ldm/mmaudio/vae/vae.py b/comfy/ldm/mmaudio/vae/vae.py index f294ec4e1..65f625238 100644 --- a/comfy/ldm/mmaudio/vae/vae.py +++ b/comfy/ldm/mmaudio/vae/vae.py @@ -75,14 +75,14 @@ class VAE(nn.Module): super().__init__() if data_dim == 80: - self.data_mean = nn.Buffer(torch.tensor(DATA_MEAN_80D, dtype=torch.float32)) - self.data_std = nn.Buffer(torch.tensor(DATA_STD_80D, dtype=torch.float32)) + data_mean = torch.tensor(DATA_MEAN_80D, dtype=torch.float32) + data_std = torch.tensor(DATA_STD_80D, dtype=torch.float32) elif data_dim == 128: - self.data_mean = nn.Buffer(torch.tensor(DATA_MEAN_128D, dtype=torch.float32)) - self.data_std = nn.Buffer(torch.tensor(DATA_STD_128D, dtype=torch.float32)) + data_mean = torch.tensor(DATA_MEAN_128D, dtype=torch.float32) + data_std = torch.tensor(DATA_STD_128D, dtype=torch.float32) - self.data_mean = self.data_mean.view(1, -1, 1) - self.data_std = self.data_std.view(1, -1, 1) + self.register_buffer('data_mean', data_mean.view(1, -1, 1)) + self.register_buffer('data_std', data_std.view(1, -1, 1)) self.encoder = Encoder1D( dim=hidden_dim, diff --git a/comfy/ops.py b/comfy/ops.py index 0f8b8b5f3..0bfad193f 100644 --- a/comfy/ops.py +++ b/comfy/ops.py @@ -31,7 +31,8 @@ logger = logging.getLogger(__name__) def run_every_op(): - if torch.compiler.is_compiling(): + # this is available on torch 2.3 + if torch.compiler.is_compiling(): # pylint: disable=no-member return throw_exception_if_processing_interrupted()