admin/scribe
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

cahnged name form dia_model to dia_dir

Browse Source
This commit is contained in:
Jaikinator
2023-06-19 16:31:42 +02:00
parent 57fd73c8ee
commit 29cc0aca6a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
autotranscript/autotranscript.py
+2 -2
View File
@@ -190,7 +190,7 @@ def cli():
parser.add_argument("--wmodel_dir", type=str, default= WHISPER_DEFAULT_PATH, parser.add_argument("--wmodel_dir", type=str, default= WHISPER_DEFAULT_PATH,
help="the path to save model files; uses ./models/whisper by default") 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, parser.add_argument("--allow_download", type= bool, default=True,
help="whether to allow model download if model is not found locally") help="whether to allow model download if model is not found locally")
@@ -237,7 +237,7 @@ def cli():
diarisation_kwargs = {"local": local} diarisation_kwargs = {"local": local}
model = AutoTranscribe(whisper_model= model_name, model = AutoTranscribe(whisper_model= model_name,
whisper_kwargs= wkwargs, whisper_kwargs= wkwargs,
dia_model= args.pop("dia_model"), dia_model= args.pop("dia_dir"),
dia_kwargs_kwargs= diarisation_kwargs,) dia_kwargs_kwargs= diarisation_kwargs,)
if task == "transcribe": if task == "transcribe":