-
Notifications
You must be signed in to change notification settings - Fork 151
Description
I am getting this issue, after runing python3 test/test.py
My pytorch version is 2.0.0+cu117
Initializing VoiceFixer...
Traceback (most recent call last):
File "/home/sughosh/voicefixer/test/test.py", line 41, in
voicefixer = VoiceFixer()
^^^^^^^^^^^^
File "/home/sughosh/miniconda3/envs/emo/lib/python3.11/site-packages/voicefixer/base.py", line 13, in init
self._model = voicefixer_fe(channels=2, sample_rate=44100)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/sughosh/miniconda3/envs/emo/lib/python3.11/site-packages/voicefixer/restorer/model.py", line 180, in init
self.vocoder = Vocoder(sample_rate=44100)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/sughosh/miniconda3/envs/emo/lib/python3.11/site-packages/voicefixer/vocoder/base.py", line 19, in init
self._load_pretrain(Config.ckpt)
File "/home/sughosh/miniconda3/envs/emo/lib/python3.11/site-packages/voicefixer/vocoder/base.py", line 25, in _load_pretrain
self.model = Generator(Config.cin_channels)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/sughosh/miniconda3/envs/emo/lib/python3.11/site-packages/voicefixer/vocoder/model/generator.py", line 34, in init
nn.utils.parametrizations.weight_norm(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: module 'torch.nn.utils.parametrizations' has no attribute 'weight_norm'