add endpoint for Qwen3-TTS -> RVC
This commit is contained in:
@@ -8,7 +8,7 @@ from tqdm import tqdm
|
||||
|
||||
|
||||
def load_inputs(path, device, is_half=False):
|
||||
parm = torch.load(path, map_location=torch.device("cpu"))
|
||||
parm = torch.load(path, map_location=torch.device("cpu"), weights_only=False)
|
||||
for key in parm.keys():
|
||||
parm[key] = parm[key].to(device)
|
||||
if is_half and parm[key].dtype == torch.float32:
|
||||
|
||||
@@ -5,7 +5,7 @@ def get_rmvpe(model_path="assets/rmvpe/rmvpe.pt", device=torch.device("cpu")):
|
||||
from infer.lib.rmvpe import E2E
|
||||
|
||||
model = E2E(4, 1, (2, 2))
|
||||
ckpt = torch.load(model_path, map_location=device)
|
||||
ckpt = torch.load(model_path, map_location=device, weights_only=False)
|
||||
model.load_state_dict(ckpt)
|
||||
model.eval()
|
||||
model = model.to(device)
|
||||
|
||||
@@ -9,7 +9,7 @@ def get_synthesizer(pth_path, device=torch.device("cpu")):
|
||||
SynthesizerTrnMs768NSFsid_nono,
|
||||
)
|
||||
|
||||
cpt = torch.load(pth_path, map_location=torch.device("cpu"))
|
||||
cpt = torch.load(pth_path, map_location=torch.device("cpu"), weights_only=False)
|
||||
# tgt_sr = cpt["config"][-1]
|
||||
cpt["config"][-3] = cpt["weight"]["emb_g.weight"].shape[0]
|
||||
if_f0 = cpt.get("f0", 1)
|
||||
|
||||
1330
rvc/lib/rmvpe.py
1330
rvc/lib/rmvpe.py
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user