From f69056d653ff0fd1faf402bcfedc1b6c53590e9b Mon Sep 17 00:00:00 2001 From: Mubarak Harran Alketbi Date: Sun, 18 Feb 2024 22:46:30 +0400 Subject: [PATCH] Update hifigan_generator.py fixing error ``` ImportError: cannot import name 'weight_norm' from 'torch.nn.utils.parametrizations' ``` module is in torch.nn.utils.weight_norm --- TTS/vocoder/models/hifigan_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TTS/vocoder/models/hifigan_generator.py b/TTS/vocoder/models/hifigan_generator.py index 92475322..e1faa4e7 100644 --- a/TTS/vocoder/models/hifigan_generator.py +++ b/TTS/vocoder/models/hifigan_generator.py @@ -3,7 +3,7 @@ import torch from torch import nn from torch.nn import Conv1d, ConvTranspose1d from torch.nn import functional as F -from torch.nn.utils.parametrizations import weight_norm +from torch.nn.utils.weight_norm import weight_norm from torch.nn.utils.parametrize import remove_parametrizations from TTS.utils.io import load_fsspec