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

Merge pull request #580 from kaka1909/main

Browse Source 
Update asr.py and make the model parameter be used
This commit is contained in:
Max Bain
2023-11-16 10:54:02 +00:00
committed by GitHub
parent 4ece2369d7 48d651e5ea
commit 7c2a9a8b7b
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,
language : Optional[str] = None,
vad_options=None,
model=None,
model : Optional[WhisperModel] = None,
task="transcribe",
download_root=None,
threads=4):
@ -40,6 +40,7 @@ def load_model(whisper_arch,
compute_type: str - The compute type 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)
model: Optional[WhisperModel] - The WhisperModel instance to use.
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.
Returns:
@ -49,7 +50,7 @@ def load_model(whisper_arch,
if whisper_arch.endswith(".en"):
language = "en"
model = WhisperModel(whisper_arch,
model = model or WhisperModel(whisper_arch,
device=device,
device_index=device_index,
compute_type=compute_type,