mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-01-24 05:55:01 +03:00
32 lines
956 B
Python
32 lines
956 B
Python
|
import torch
|
||
|
|
from torch import device
|
||
|
|
|
||
|
|
from const import EnumInferenceTypes
|
||
|
|
from voice_changer.RVC.inferencer.Inferencer import Inferencer
|
||
|
|
from .models import SynthesizerTrnMsNSFsidNono
|
||
|
|
|
||
|
|
|
||
|
|
class WebUIInferencerNono(Inferencer):
|
||
|
|
def loadModel(self, file: str, dev: device, isHalf: bool = True):
|
||
|
|
super().setProps(EnumInferenceTypes.pyTorchRVC, file, dev, isHalf)
|
||
|
|
cpt = torch.load(file, map_location="cpu")
|
||
|
|
model = SynthesizerTrnMsNSFsidNono(**cpt["params"], is_half=isHalf)
|
||
|
|
|
||
|
|
model.eval()
|
||
|
|
model.load_state_dict(cpt["weight"], strict=False)
|
||
|
|
if isHalf:
|
||
|
|
model = model.half()
|
||
|
|
|
||
|
|
self.model = model
|
||
|
|
return self
|
||
|
|
|
||
|
|
def infer(
|
||
|
|
self,
|
||
|
|
feats: torch.Tensor,
|
||
|
|
pitch_length: torch.Tensor,
|
||
|
|
pitch: torch.Tensor | None,
|
||
|
|
pitchf: torch.Tensor | None,
|
||
|
|
sid: torch.Tensor,
|
||
|
|
) -> torch.Tensor:
|
||
|
|
return self.model.infer(feats, pitch_length, sid)
|