fix wavernn

mockingbirdforuse/vocoder/wavernn/inference.py

@@ -36,13 +36,7 @@ def __init__(self, model_path: Path):
         self._model.eval()
 
     def infer_waveform(
-        self,
-        mel,
-        normalize=True,
-        batched=True,
-        target=8000,
-        overlap=800,
-        progress_callback=None,
+        self, mel, normalize=True, batched=True, target=8000, overlap=800
     ):
         """
         Infers the waveform of a mel spectrogram output by the synthesizer (the format must match
@@ -58,7 +52,5 @@ def infer_waveform(
         if normalize:
             mel = mel / hp.mel_max_abs_value
         mel = torch.from_numpy(mel[None, ...])
-        wav = self._model.generate(
-            mel, batched, target, overlap, hp.mu_law, progress_callback
-        )
+        wav = self._model.generate(mel, batched, target, overlap, hp.mu_law)
         return wav, hp.sample_rate

mockingbirdforuse/vocoder/wavernn/models/fatchord_version.py

@@ -172,9 +172,8 @@ def forward(self, x, mels):
         x = F.relu(self.fc2(x))
         return self.fc3(x)
 
-    def generate(self, mels, batched, target, overlap, mu_law, progress_callback=None):
+    def generate(self, mels, batched, target, overlap, mu_law):
         mu_law = mu_law if self.mode == "RAW" else False
-        progress_callback = progress_callback or self.gen_display
 
         self.eval()
         output = []

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "MockingBirdForUse"
-version = "0.2.1"
+version = "0.2.2"
 description = ""
 authors = ["meetwq <[email protected]>"]