From 81e61dccd2aca22c31523ebfa701342fb9d91712 Mon Sep 17 00:00:00 2001 From: Priyanshu Thakare Date: Thu, 2 Apr 2026 10:53:52 +0530 Subject: [PATCH] Fix CPU compute type for whisper inference --- tribev2/eventstransforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tribev2/eventstransforms.py b/tribev2/eventstransforms.py index 1fd0c69..4e4444b 100644 --- a/tribev2/eventstransforms.py +++ b/tribev2/eventstransforms.py @@ -105,7 +105,7 @@ def _get_transcript_from_audio(wav_filename: Path, language: str) -> pd.DataFram raise ValueError(f"Language {language} not supported") device = "cuda" if torch.cuda.is_available() else "cpu" - compute_type = "float16" + compute_type = "float16" if device == "cuda" else "int8" with tempfile.TemporaryDirectory() as output_dir: logger.info("Running whisperx via uvx...")