From 29cc0aca6ad72b243396d2a52196714bac016ae0 Mon Sep 17 00:00:00 2001 From: Jaikinator Date: Mon, 19 Jun 2023 16:31:42 +0200 Subject: [PATCH] cahnged name form dia_model to dia_dir --- autotranscript/autotranscript.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/autotranscript/autotranscript.py b/autotranscript/autotranscript.py index 2097f2f..42ed015 100644 --- a/autotranscript/autotranscript.py +++ b/autotranscript/autotranscript.py @@ -190,7 +190,7 @@ def cli(): parser.add_argument("--wmodel_dir", type=str, default= WHISPER_DEFAULT_PATH, help="the path to save model files; uses ./models/whisper by default") - parser.add_argument("--dia_model", type=str, default = PYANNOTE_DEFAULT_PATH) + parser.add_argument("--dia_dir", type=str, default = PYANNOTE_DEFAULT_PATH) parser.add_argument("--allow_download", type= bool, default=True, help="whether to allow model download if model is not found locally") @@ -237,7 +237,7 @@ def cli(): diarisation_kwargs = {"local": local} model = AutoTranscribe(whisper_model= model_name, whisper_kwargs= wkwargs, - dia_model= args.pop("dia_model"), + dia_model= args.pop("dia_dir"), dia_kwargs_kwargs= diarisation_kwargs,) if task == "transcribe":