1
0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-07 10:10:23 +08:00

feat: update to latest torch & gradio version

This commit is contained in:
源文雨
2024-11-27 22:16:06 +09:00
parent 7c0b1c01f1
commit 4b68fb0e13
8 changed files with 40 additions and 37 deletions

View File

@@ -7,8 +7,9 @@ def get_rmvpe(
from rvc.f0.e2e 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=True)
model.load_state_dict(ckpt)
del ckpt
model.eval()
if is_half:
model = model.half()