diff options
| author | yum <yum.food.vr@gmail.com> | 2024-02-09 17:51:53 -0800 |
|---|---|---|
| committer | yum <yum.food.vr@gmail.com> | 2024-02-09 17:51:53 -0800 |
| commit | 5ef207d28f2a9d943384b9ec6872aedae2917ac0 (patch) | |
| tree | 9f50b4bd5bc3a40e8c79b0383060638adade864a /Scripts | |
| parent | 3b84b185d1286e1b954f5ad636b26188efa141e4 (diff) | |
Finish plumbing GPU compute typev0.18.1
Diffstat (limited to 'Scripts')
| -rw-r--r-- | Scripts/transcribe_v2.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/Scripts/transcribe_v2.py b/Scripts/transcribe_v2.py index aec8729..7655d4b 100644 --- a/Scripts/transcribe_v2.py +++ b/Scripts/transcribe_v2.py @@ -435,7 +435,7 @@ class Whisper: self.model = WhisperModel(model_str, device = model_device, device_index = cfg["gpu_idx"], - compute_type = "float16", + compute_type = cfg["compute_type"], download_root = model_root, local_files_only = download_it) |