heimoshuiyu/faster-whisper
1
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

Fix typing for device_index argument

Browse Source
This commit is contained in:
Guillaume Klein
2023-03-22 13:51:12 +01:00
parent c27c010f96
commit 52264f2277
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
faster_whisper/transcribe.py
View File

@@ -59,7 +59,7 @@ class WhisperModel:
self, self,
model_path: str, model_path: str,
device: str = "auto", device: str = "auto",
device_index: int = 0, device_index: Union[int, List[int]] = 0,
compute_type: str = "default", compute_type: str = "default",
cpu_threads: int = 0, cpu_threads: int = 0,
num_workers: int = 1, num_workers: int = 1,