File size: 372 Bytes
c7b379a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import torch
def get_rmvpe(
model_path="assets/rmvpe/rmvpe.pt", device=torch.device("cpu"), is_half=False
):
from rvc.f0.e2e import E2E
model = E2E(4, 1, (2, 2))
ckpt = torch.load(model_path, map_location=device)
model.load_state_dict(ckpt)
model.eval()
if is_half:
model = model.half()
model = model.to(device)
return model
|