NeoPy commited on
Commit
a0422fb
·
verified ·
1 Parent(s): d487e44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -10,7 +10,11 @@ import torch
10
  import asyncio
11
  import edge_tts
12
  from datetime import datetime
13
- from fairseq import checkpoint_utils
 
 
 
 
14
  from infer_pack.models import SynthesizerTrnMs256NSFsid, SynthesizerTrnMs256NSFsid_nono
15
  from vc_infer_pipeline import VC
16
  from config import (
@@ -80,10 +84,9 @@ def create_vc_fn(tgt_sr, net_g, vc, if_f0, file_index, file_big_npy):
80
 
81
  def load_hubert():
82
  global hubert_model
83
- models, _, _ = checkpoint_utils.load_model_ensemble_and_task(
84
- ["hubert_base.pt"],
85
- suffix="",
86
- )
87
  hubert_model = models[0]
88
  hubert_model = hubert_model.to(device)
89
  if is_half:
 
10
  import asyncio
11
  import edge_tts
12
  from datetime import datetime
13
+ from fairseq.fairseq import load_model
14
+
15
+
16
+
17
+
18
  from infer_pack.models import SynthesizerTrnMs256NSFsid, SynthesizerTrnMs256NSFsid_nono
19
  from vc_infer_pipeline import VC
20
  from config import (
 
84
 
85
  def load_hubert():
86
  global hubert_model
87
+ models = load_model("hubert_base.pt")
88
+
89
+
 
90
  hubert_model = models[0]
91
  hubert_model = hubert_model.to(device)
92
  if is_half: