ImportError: cannot import name 'consume_prefix_in_state_dict_if_present' from 'torch.nn.modules.utils'

quickvc / QuickVC-VoiceConversion

QuickVC: Any-to-many Voice Conversion Using Inverse Short-time Fourier Transform for Faster Conversion

MIT License

225 stars 26 forks source link

ImportError: cannot import name 'consume_prefix_in_state_dict_if_present' from 'torch.nn.modules.utils' #22

Open zxj329 opened 3 weeks ago

zxj329 commented 3 weeks ago

加载hubert-soft模型的时候，hubert = torch.hub.load("bshall/hubert:main", f"hubert_soft").to(device).eval()报了这个错误，我的torch版本是1.8.0+cu111，请问下需要哪个torch版本。如果使用torch2.x以上的版本，你代码中的stft相关的会报错。很矛盾的事情