From 69ece5b811f766c1cae23f7493d0d4bca7ae5efa Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 6 Oct 2022 21:08:20 +0530 Subject: [PATCH] to numpy --- enhancer/inference.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/enhancer/inference.py b/enhancer/inference.py index fd3b518..d9282fd 100644 --- 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(