diff options
| author | yum-food <114886918+yum-food@users.noreply.github.com> | 2023-06-30 00:15:50 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-06-30 00:15:50 -0700 |
| commit | ab1d4499e1b3198b387a1e2d7333a93f694cdfae (patch) | |
| tree | 06d8cc224205ff7c6a989063f05f2960f1256001 /Scripts/transcribe.py | |
| parent | 3b10d3ab3073af2ed716d1607bb92394bb8817fc (diff) | |
| parent | 95927745b3d3b2ac566f1cfd634a40e760ed29cb (diff) | |
Merge pull request #3 from jsopn/fix-gpu-device-indexv0.13.0
Set GPU device index in whisper model
Diffstat (limited to 'Scripts/transcribe.py')
| -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) |