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

Added docstring.

Browse Source
This commit is contained in:
Marko Henning
2024-05-13 11:46:01 +02:00
parent 82e26771e0
commit f4bf85b7d1
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scraibe/autotranscript.py
+2
View File
@@ -72,6 +72,8 @@ class Scraibe:
Args: Args:
whisper_model (Union[bool, str, whisper], optional): whisper_model (Union[bool, str, whisper], optional):
Path to whisper model or whisper model itself. 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): diarisation_model (Union[bool, str, DiarisationType], optional):
Path to pyannote diarization model or model itself. Path to pyannote diarization model or model itself.
**kwargs: Additional keyword arguments for whisper **kwargs: Additional keyword arguments for whisper
scraibe/transcriber.py
+2 -2
View File
@@ -140,10 +140,10 @@ class Transcriber:
- 'large-v2' - 'large-v2'
- 'large-v3' - 'large-v3'
- 'large' - 'large'
whisper_type (str):
Type of whisper model to load. "whisper" or "whisperx".
download_root (str, optional): Path to download the model. download_root (str, optional): Path to download the model.
Defaults to WHISPER_DEFAULT_PATH. Defaults to WHISPER_DEFAULT_PATH.
device (Optional[Union[str, torch.device]], optional): device (Optional[Union[str, torch.device]], optional):
Device to load model on. Defaults to None. Device to load model on. Defaults to None.
in_memory (bool, optional): Whether to load model in memory. in_memory (bool, optional): Whether to load model in memory.