diff options
| author | jsopn <github@jsopn.com> | 2023-06-30 07:15:30 +0700 |
|---|---|---|
| committer | jsopn <github@jsopn.com> | 2023-06-30 07:15:30 +0700 |
| commit | 95927745b3d3b2ac566f1cfd634a40e760ed29cb (patch) | |
| tree | 06d8cc224205ff7c6a989063f05f2960f1256001 /Scripts | |
| parent | 3b10d3ab3073af2ed716d1607bb92394bb8817fc (diff) | |
fix: set gpu device index in whisper model
Diffstat (limited to 'Scripts')
| -rw-r--r-- | Scripts/transcribe.py | 1 |
1 files changed, 1 insertions, 0 deletions
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) |