From 5ef207d28f2a9d943384b9ec6872aedae2917ac0 Mon Sep 17 00:00:00 2001
From: yum <yum.food.vr@gmail.com>
Date: Fri, 9 Feb 2024 17:51:53 -0800
Subject: Finish plumbing GPU compute type

---
 Scripts/transcribe_v2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)
 
-- 
cgit v1.2.3