r3gm commited on
Commit
9cd58c5
·
verified ·
1 Parent(s): 9bfa452

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -23,6 +23,8 @@ import shutil
23
  import threading
24
  import argparse
25
  import sys
 
 
26
 
27
  parser = argparse.ArgumentParser(description="Run the app with optional sharing")
28
  parser.add_argument(
@@ -43,6 +45,7 @@ IS_ZERO_GPU = os.getenv("SPACES_ZERO_GPU")
43
 
44
  logging.getLogger("infer_rvc_python").setLevel(logging.ERROR)
45
 
 
46
  converter = BaseLoader(only_cpu=False, hubert_path=None, rmvpe_path=None)
47
  converter.hu_bert_model = load_hu_bert(Config(only_cpu=False), converter.hubert_path)
48
 
 
23
  import threading
24
  import argparse
25
  import sys
26
+ import torch
27
+ import fairseq
28
 
29
  parser = argparse.ArgumentParser(description="Run the app with optional sharing")
30
  parser.add_argument(
 
45
 
46
  logging.getLogger("infer_rvc_python").setLevel(logging.ERROR)
47
 
48
+ torch.serialization.add_safe_globals([fairseq.data.dictionary.Dictionary])
49
  converter = BaseLoader(only_cpu=False, hubert_path=None, rmvpe_path=None)
50
  converter.hu_bert_model = load_hu_bert(Config(only_cpu=False), converter.hubert_path)
51