diff --git a/infer/modules/train/extract_f0_print.py b/infer/modules/train/extract_f0_print.py index d0c3c03..243606c 100644 --- a/infer/modules/train/extract_f0_print.py +++ b/infer/modules/train/extract_f0_print.py @@ -40,7 +40,8 @@ device = sys.argv[4] is_half = sys.argv[5] == "True" if importlib.util.find_spec("torch_directml") is not None: - import torch_directml # use side effect + import torch_directml # use side effect + class FeatureInput(object): def __init__(self, is_half: bool, device="cpu", samplerate=16000, hop_size=160): diff --git a/rvc/layers/synthesizers.py b/rvc/layers/synthesizers.py index 2bb7ea4..c2c70ba 100644 --- a/rvc/layers/synthesizers.py +++ b/rvc/layers/synthesizers.py @@ -164,7 +164,7 @@ class SynthesizerTrnMsNSFsid(nn.Module): ) if pitchf is not None and isinstance(self.dec, NSFGenerator): pitchf = slice_on_last_dim(pitchf, ids_slice, self.segment_size) - o = self.dec(z_slice, pitchf, g=embg) # type: ignore + o = self.dec(z_slice, pitchf, g=embg) # type: ignore elif isinstance(self.dec, Generator): o = self.dec(z_slice, g=embg) else: @@ -204,7 +204,7 @@ class SynthesizerTrnMsNSFsid(nn.Module): z = self.flow(z_p, x_mask, g=g, reverse=True) del z_p, m_p, logs_p if pitchf is not None and isinstance(self.dec, NSFGenerator): - o = self.dec( + o = self.dec( z * x_mask, pitchf, g=g,