13 lines
311 B
Python
13 lines
311 B
Python
|
import torch
|
||
|
|
|
||
|
|
|
||
|
|
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)
|
||
|
|
model.load_state_dict(ckpt)
|
||
|
|
model.eval()
|
||
|
|
model = model.to(device)
|
||
|
|
return model
|