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

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

diff --git a/infer/modules/vc/modules.py b/infer/modules/vc/modules.py
index 96db6a7..2e68fc3 100644
--- a/infer/modules/vc/modules.py
+++ b/infer/modules/vc/modules.py
@@ -20,6 +20,12 @@ 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)
+
 class VC:
     def __init__(self, config):
         self.n_spk = None
@@ -170,8 +176,7 @@ class VC:
             times = [0, 0, 0]
 
             if self.hubert_model is None:
-                with torch.serialization.safe_globals({"fairseq.data.dictionary.Dictionary": Dictionary}):
-                    self.hubert_model = load_hubert(self.config)
+                self.hubert_model = load_hubert_with_safe_globals(self.config)
 
             if file_index:
                 file_index = (