to numpy
This commit is contained in:
parent
2e0a6b636b
commit
69ece5b811
|
|
@ -139,7 +139,9 @@ class Inference:
|
||||||
if filename.is_file():
|
if filename.is_file():
|
||||||
raise FileExistsError(f"file {filename} already exists")
|
raise FileExistsError(f"file {filename} already exists")
|
||||||
else:
|
else:
|
||||||
wavfile.write(filename, rate=sr, data=waveform.detach().cpu())
|
wavfile.write(
|
||||||
|
filename, rate=sr, data=waveform.detach().cpu().numpy()
|
||||||
|
)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def prepare_output(
|
def prepare_output(
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue