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-04 03:38:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

hubert

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

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

@ -20,11 +20,7 @@ from infer.modules.vc.utils import *
from fairseq.data.dictionary import Dictionary
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)
torch.serialization.add_safe_globals([Dictionary])
class VC:
def __init__(self, config):
@ -176,7 +172,7 @@ class VC:
times = [0, 0, 0]
if self.hubert_model is None:
self.hubert_model = load_hubert_with_safe_globals(self.config)
self.hubert_model = load_hubert(self.config)
if file_index:
file_index = (