Added docstring.

2024-05-13 11:46:01 +02:00
parent 82e26771e0
commit f4bf85b7d1
2 changed files with 4 additions and 2 deletions
@@ -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.