From 8ddc151a4ce7f59de75b24ea8349f0e406ba0da5 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Sun, 16 Jun 2024 13:06:23 -0400 Subject: [PATCH] Squash depreciation warning on new pytorch. --- comfy/ldm/audio/autoencoder.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/comfy/ldm/audio/autoencoder.py b/comfy/ldm/audio/autoencoder.py index 7363131e..8123e66a 100644 --- a/comfy/ldm/audio/autoencoder.py +++ b/comfy/ldm/audio/autoencoder.py @@ -75,10 +75,16 @@ class SnakeBeta(nn.Module): return x def WNConv1d(*args, **kwargs): - return torch.nn.utils.weight_norm(ops.Conv1d(*args, **kwargs)) + try: + return torch.nn.utils.parametrizations.weight_norm(ops.Conv1d(*args, **kwargs)) + except: + return torch.nn.utils.weight_norm(ops.Conv1d(*args, **kwargs)) #support pytorch 2.1 and older def WNConvTranspose1d(*args, **kwargs): - return torch.nn.utils.weight_norm(ops.ConvTranspose1d(*args, **kwargs)) + try: + return torch.nn.utils.parametrizations.weight_norm(ops.ConvTranspose1d(*args, **kwargs)) + except: + return torch.nn.utils.weight_norm(ops.ConvTranspose1d(*args, **kwargs)) #support pytorch 2.1 and older def get_activation(activation: Literal["elu", "snake", "none"], antialias=False, channels=None) -> nn.Module: if activation == "elu":