Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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:
Mubarak Harran Alketbi 2024-02-18 22:46:30 +04:00 committed by GitHub
parent dbf1a08a0d
commit f69056d653
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/vocoder/models/hifigan_generator.py
View File

@ -3,7 +3,7 @@ import torch
from torch import nn from torch import nn
from torch.nn import Conv1d, ConvTranspose1d from torch.nn import Conv1d, ConvTranspose1d
from torch.nn import functional as F 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 torch.nn.utils.parametrize import remove_parametrizations
from TTS.utils.io import load_fsspec from TTS.utils.io import load_fsspec