Simplified fix for encode_mp3

mengzhisuoliu · Jul 19, 2022 · 1c9b9ab · 1c9b9ab
1 parent ce9a04d
commit 1c9b9ab
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/demucs/audio.py b/demucs/audio.py
@@ -210,8 +210,7 @@ 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.data.cpu()
     wav = wav.transpose(0, 1).numpy()
     mp3_data = encoder.encode(wav.tobytes())
     mp3_data += encoder.flush()