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

Update modules/speaker.py

Browse files
Files changed (1) hide show
  1. modules/speaker.py +2 -1
modules/speaker.py CHANGED
@@ -26,7 +26,8 @@ def create_speaker_from_seed(seed):
26
  class Speaker:
27
  @staticmethod
28
  def from_file(file_like):
29
- speaker = torch.load(file_like, map_location=torch.device("cpu"))
 
30
  speaker.fix()
31
  return speaker
32
 
 
26
  class Speaker:
27
  @staticmethod
28
  def from_file(file_like):
29
+ with safe_globals({'modules.speaker.Speaker': Speaker}):
30
+ speaker = torch.load(file_like, map_location=torch.device("cpu"))
31
  speaker.fix()
32
  return speaker
33