Pass device to demucs to ensure correct device is used (#284)

MahmoudAshraf97 · Dec 15, 2024 · 2d0745e · 2d0745e
1 parent f35796c
commit 2d0745e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/Whisper_Transcription_+_NeMo_Diarization.ipynb b/Whisper_Transcription_+_NeMo_Diarization.ipynb
@@ -765,7 +765,7 @@
     "    # Isolate vocals from the rest of the audio\n",
     "\n",
     "    return_code = os.system(\n",
-    "        f'python -m demucs.separate -n htdemucs --two-stems=vocals \"{audio_path}\" -o \"temp_outputs\"'\n",
+    "        f'python -m demucs.separate -n htdemucs --two-stems=vocals \"{audio_path}\" -o \"temp_outputs\" --device \"{device}\"'\n",
     "    )\n",
     "\n",
     "    if return_code != 0:\n",

diff --git a/diarize.py b/diarize.py
@@ -96,7 +96,7 @@
     # Isolate vocals from the rest of the audio
 
     return_code = os.system(
-        f'python -m demucs.separate -n htdemucs --two-stems=vocals "{args.audio}" -o temp_outputs'
+        f'python -m demucs.separate -n htdemucs --two-stems=vocals "{args.audio}" -o temp_outputs --device "{args.device}"'
     )
 
     if return_code != 0:

diff --git a/diarize_parallel.py b/diarize_parallel.py
@@ -94,7 +94,7 @@
     # Isolate vocals from the rest of the audio
 
     return_code = os.system(
-        f'python -m demucs.separate -n htdemucs --two-stems=vocals "{args.audio}" -o temp_outputs'
+        f'python -m demucs.separate -n htdemucs --two-stems=vocals "{args.audio}" -o temp_outputs --device "{args.device}"'
     )
 
     if return_code != 0: