add endpoint for Qwen3-TTS -> RVC

This commit is contained in:
2026-03-01 15:32:48 -08:00
parent 7b284a6346
commit 1f9832ac0d
15 changed files with 1199 additions and 3337 deletions

View File

@@ -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:

View File

@@ -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)

View File

@@ -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)

File diff suppressed because it is too large Load Diff