From f4bf85b7d14a589ccc20d627d6e84954b1fe7c63 Mon Sep 17 00:00:00 2001 From: Marko Henning Date: Mon, 13 May 2024 11:46:01 +0200 Subject: [PATCH] Added docstring. --- scraibe/autotranscript.py | 2 ++ scraibe/transcriber.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/scraibe/autotranscript.py b/scraibe/autotranscript.py index 4081638..cf77a62 100644 --- a/scraibe/autotranscript.py +++ b/scraibe/autotranscript.py @@ -72,6 +72,8 @@ class Scraibe: Args: whisper_model (Union[bool, str, whisper], optional): Path to whisper model or whisper model itself. + whisper_type (str): + Type of whisper model to load. "whisper" or "whisperx". diarisation_model (Union[bool, str, DiarisationType], optional): Path to pyannote diarization model or model itself. **kwargs: Additional keyword arguments for whisper diff --git a/scraibe/transcriber.py b/scraibe/transcriber.py index 365d321..977cd94 100644 --- a/scraibe/transcriber.py +++ b/scraibe/transcriber.py @@ -140,10 +140,10 @@ class Transcriber: - 'large-v2' - 'large-v3' - 'large' - + whisper_type (str): + Type of whisper model to load. "whisper" or "whisperx". download_root (str, optional): Path to download the model. Defaults to WHISPER_DEFAULT_PATH. - device (Optional[Union[str, torch.device]], optional): Device to load model on. Defaults to None. in_memory (bool, optional): Whether to load model in memory.