OPENGIT/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git synced 2025-04-05 04:08:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Hubert

Browse Source
This commit is contained in:
VSlobolinskyi 2025-03-18 11:11:31 +02:00
parent fbda348b9a
commit dac00219c3
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
infer/modules/vc/modules.py
View File

@ -20,6 +20,12 @@ from infer.modules.vc.utils import *
from fairseq.data.dictionary import Dictionary from fairseq.data.dictionary import Dictionary
import torch import torch
def load_hubert_with_safe_globals(config):
safe_globals = {"fairseq.data.dictionary.Dictionary": Dictionary}
# Wrap the loading call in the safe_globals context manager.
with torch.serialization.safe_globals(safe_globals):
return load_hubert(config)
class VC: class VC:
def __init__(self, config): def __init__(self, config):
self.n_spk = None self.n_spk = None
@ -170,8 +176,7 @@ class VC:
times = [0, 0, 0] times = [0, 0, 0]
if self.hubert_model is None: if self.hubert_model is None:
with torch.serialization.safe_globals({"fairseq.data.dictionary.Dictionary": Dictionary}): self.hubert_model = load_hubert_with_safe_globals(self.config)
self.hubert_model = load_hubert(self.config)
if file_index: if file_index:
file_index = ( file_index = (