ProbablyBlue commited on
Commit
7791837
·
verified ·
1 Parent(s): 10a102f

Update modules/speaker.py

Browse files
Files changed (1) hide show
  1. modules/speaker.py +7 -0
modules/speaker.py CHANGED
@@ -8,7 +8,14 @@ from box import Box
8
  from modules import models
9
  from modules.utils.SeedContext import SeedContext
10
 
 
 
11
 
 
 
 
 
 
12
  def create_speaker_from_seed(seed):
13
  chat_tts = models.load_chat_tts()
14
  with SeedContext(seed, True):
 
8
  from modules import models
9
  from modules.utils.SeedContext import SeedContext
10
 
11
+ from torch.serialization import add_safe_globals
12
+ from modules.speaker import Speaker
13
 
14
+ add_safe_globals({'modules.speaker.Speaker': Speaker})
15
+
16
+ with safe_globals({'modules.speaker.Speaker': Speaker}):
17
+ speaker = torch.load(file_like, map_location="cpu", weights_only=False)
18
+
19
  def create_speaker_from_seed(seed):
20
  chat_tts = models.load_chat_tts()
21
  with SeedContext(seed, True):