From 2291e77f53fe3c54008d07b10b439c02f67ea8ba Mon Sep 17 00:00:00 2001 From: Marko Henning Date: Fri, 17 May 2024 11:55:23 +0200 Subject: [PATCH] Actually use the new compute type variable... --- scraibe/transcriber.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scraibe/transcriber.py b/scraibe/transcriber.py index 526ff8b..0301955 100644 --- a/scraibe/transcriber.py +++ b/scraibe/transcriber.py @@ -348,7 +348,7 @@ class WhisperXTranscriber(Transcriber): f'device {device}! Changing compute type to int8.') compute_type = 'int8' _model = whisperx_load_model(model, download_root=download_root, - device=device) + device=device, compute_type=compute_type) return cls(_model, model_name=model)