Actually use the new compute type variable...

This commit is contained in:
Marko Henning
2024-05-17 11:55:23 +02:00
parent b6bed3ebd8
commit 2291e77f53
+1 -1
View File
@@ -348,7 +348,7 @@ class WhisperXTranscriber(Transcriber):
f'device {device}! Changing compute type to int8.') f'device {device}! Changing compute type to int8.')
compute_type = 'int8' compute_type = 'int8'
_model = whisperx_load_model(model, download_root=download_root, _model = whisperx_load_model(model, download_root=download_root,
device=device) device=device, compute_type=compute_type)
return cls(_model, model_name=model) return cls(_model, model_name=model)