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.