From 95927745b3d3b2ac566f1cfd634a40e760ed29cb Mon Sep 17 00:00:00 2001
From: jsopn <github@jsopn.com>
Date: Fri, 30 Jun 2023 07:15:30 +0700
Subject: fix: set gpu device index in whisper model

---
 Scripts/transcribe.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Scripts/transcribe.py b/Scripts/transcribe.py
index a6fbe38..ffebf20 100644
--- a/Scripts/transcribe.py
+++ b/Scripts/transcribe.py
@@ -755,6 +755,7 @@ def transcribeLoop(mic: str,
         model = model_root
     model = WhisperModel(model,
             device = model_device,
+            device_index = gpu_idx,
             compute_type = "int8",
             download_root = model_root,
             local_files_only = download_it)
-- 
cgit v1.2.3