From fd0487d92b78618db41cffc290d14b3f67ff5387 Mon Sep 17 00:00:00 2001
From: VSlobolinskyi <vitalii.slobolinskyi@gmail.com>
Date: Tue, 18 Mar 2025 11:16:38 +0200
Subject: [PATCH] hubert

---
 infer/modules/vc/modules.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/infer/modules/vc/modules.py b/infer/modules/vc/modules.py
index 2e68fc3..36eef0b 100644
--- a/infer/modules/vc/modules.py
+++ b/infer/modules/vc/modules.py
@@ -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 = (