mirror of https://github.com/coqui-ai/TTS.git
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
This commit is contained in:
parent
dbf1a08a0d
commit
f69056d653
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue