From 428a970c15e259ce1eb729f6a1bd967b0a3ba012 Mon Sep 17 00:00:00 2001
From: Arpan Tripathi <42506819+tripathiarpan20@users.noreply.github.com>
Date: Sun, 10 Jul 2022 13:12:04 +0530
Subject: [PATCH] minor device bug fixed

---
 demucs/audio.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/demucs/audio.py b/demucs/audio.py
index d1ba194e..5e01a8cd 100644
--- 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()