From 2a29f0ec6a8df93152949053db2fe471ea6e16cc Mon Sep 17 00:00:00 2001 From: Max Bain Date: Mon, 24 Apr 2023 21:24:22 +0100 Subject: [PATCH] add compute types --- whisperx/transcribe.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/whisperx/transcribe.py b/whisperx/transcribe.py index c3719ba..0c6b803 100644 --- a/whisperx/transcribe.py +++ b/whisperx/transcribe.py @@ -22,6 +22,8 @@ def cli(): parser.add_argument("--model_dir", type=str, default=None, help="the path to save model files; uses ~/.cache/whisper by default") parser.add_argument("--device", default="cuda" if torch.cuda.is_available() else "cpu", help="device to use for PyTorch inference") parser.add_argument("--batch_size", default=8, type=int, help="device to use for PyTorch inference") + parser.add_argument("--compute_type", default="float16", type=str, choices=["float16", "float32", "int8"], help="compute type for computation") + parser.add_argument("--output_dir", "-o", type=str, default=".", help="directory to save the outputs") parser.add_argument("--output_format", "-f", type=str, default="all", choices=["all", "srt", "vtt", "txt", "tsv", "json"], help="format of the output file; if not specified, all available formats will be produced")