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...")