vyles commited on
Commit
5f27a69
·
verified ·
1 Parent(s): efe211d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -77,11 +77,11 @@ def load_model(sovits_path, gpt_path):
77
  break
78
 
79
  if dict_s2 is None:
80
- dict_s2 = torch.load(sovits_path, map_location="cpu")
81
  hps = dict_s2["config"]
82
 
83
  if dict_s1 is None:
84
- dict_s1 = torch.load(gpt_path, map_location="cpu")
85
  config = dict_s1["config"]
86
  class DictToAttrRecursive:
87
  def __init__(self, input_dict):
 
77
  break
78
 
79
  if dict_s2 is None:
80
+ dict_s2 = torch.load(sovits_path, map_location="cpu", weights_only=False)
81
  hps = dict_s2["config"]
82
 
83
  if dict_s1 is None:
84
+ dict_s1 = torch.load(gpt_path, map_location="cpu", weights_only=False)
85
  config = dict_s1["config"]
86
  class DictToAttrRecursive:
87
  def __init__(self, input_dict):