Fix typing for device_index argument

2023-03-22 13:51:12 +01:00
parent c27c010f96
commit 52264f2277
1 changed files with 1 additions and 1 deletions
--- a/faster_whisper/transcribe.py
+++ b/faster_whisper/transcribe.py
@@ -59,7 +59,7 @@ class WhisperModel:
        self,
        model_path: str,
        device: str = "auto",
-        device_index: int = 0,
+        device_index: Union[int, List[int]] = 0,
        compute_type: str = "default",
        cpu_threads: int = 0,
        num_workers: int = 1,