configures device type for model usage

ClubCedille · Jul 19, 2024 · 3907273 · 3907273
1 parent 36c59f4
commit 3907273
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/point.py b/point.py
@@ -24,7 +24,7 @@ def transcribe(audio_file: io.BytesIO) -> Iterator[str]:
 
     # TODO: configurable model path
     whisper = WhisperModel(
-        model_size, download_root="./whisper-models", compute_type="int8"
+        model_size, download_root="./whisper-models", device="cpu"
     )
     segments, _ = whisper.transcribe(
         audio_file,