Merge pull request facebookresearch#345 from tripathiarpan20/main

minor device bug fixed with --no-split option
mengzhisuoliu · Jul 19, 2022 · ce9a04d · ce9a04d
2 parents 83f2ea8 + 428a970
commit ce9a04d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/demucs/audio.py b/demucs/audio.py
@@ -210,6 +210,8 @@ def encode_mp3(wav, path, samplerate=44100, bitrate=320, verbose=False):
     encoder.set_quality(2)  # 2-highest, 7-fastest
     if not verbose:
         encoder.silence()
+    if 'cuda' in str(wav.device):
+        wav = wav.cpu()
     wav = wav.transpose(0, 1).numpy()
     mp3_data = encoder.encode(wav.tobytes())
     mp3_data += encoder.flush()