to numpy

2022-10-06 21:08:20 +05:30 · 2022-10-06 21:08:20 +05:30 · 69ece5b811
parent 2e0a6b636b
commit 69ece5b811
1 changed files with 3 additions and 1 deletions
--- a/enhancer/inference.py
+++ b/enhancer/inference.py
@ -139,7 +139,9 @@ class Inference:
        if filename.is_file():
            raise FileExistsError(f"file {filename} already exists")
        else:
-            wavfile.write(filename, rate=sr, data=waveform.detach().cpu())
+            wavfile.write(
+                filename, rate=sr, data=waveform.detach().cpu().numpy()
+            )

    @staticmethod
    def prepare_output(