chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update asr.py and make the model parameter be used

Browse Source
This commit is contained in:
kaka1909
2023-11-16 15:29:24 +08:00
committed by GitHub
parent 4ece2369d7
commit 48d651e5ea
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
whisperx/asr.py
View File

@ -29,7 +29,7 @@ def load_model(whisper_arch,
asr_options=None, asr_options=None,
language : Optional[str] = None, language : Optional[str] = None,
vad_options=None, vad_options=None,
model=None, model : Optional[WhisperModel] = None,
task="transcribe", task="transcribe",
download_root=None, download_root=None,
threads=4): threads=4):
@ -40,6 +40,7 @@ def load_model(whisper_arch,
compute_type: str - The compute type to use for the model. compute_type: str - The compute type to use for the model.
options: dict - A dictionary of options to use for the model. options: dict - A dictionary of options to use for the model.
language: str - The language of the model. (use English for now) language: str - The language of the model. (use English for now)
model: Optional[WhisperModel] - The WhisperModel instance to use.
download_root: Optional[str] - The root directory to download the model to. download_root: Optional[str] - The root directory to download the model to.
threads: int - The number of cpu threads to use per worker, e.g. will be multiplied by num workers. threads: int - The number of cpu threads to use per worker, e.g. will be multiplied by num workers.
Returns: Returns:
@ -49,7 +50,7 @@ def load_model(whisper_arch,
if whisper_arch.endswith(".en"): if whisper_arch.endswith(".en"):
language = "en" language = "en"
model = WhisperModel(whisper_arch, model = model or WhisperModel(whisper_arch,
device=device, device=device,
device_index=device_index, device_index=device_index,
compute_type=compute_type, compute_type=compute_type,