vlad.os/whisperx-rocm-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: remove redundant variable & improve load_model function documentation (#1197)

Browse Source 
* Remove redundant variable

* Improve function documentation
This commit is contained in:
3manifold
2025-10-09 09:32:02 +02:00
committed by GitHub
parent 2663f2edb5
commit 64e307cc29
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
whisperx/asr.py
View File

@@ -319,7 +319,8 @@ def load_model(
whisper_arch - The name of the Whisper model to load. whisper_arch - The name of the Whisper model to load.
device - The device to load the model on. device - The device to load the model on.
compute_type - The compute type to use for the model. compute_type - The compute type to use for the model.
vad_method - The vad method to use. vad_model has higher priority if is not None. vad_model - The vad model to manually assign.
vad_method - The vad method to use. vad_model has a higher priority if it is not None.
options - A dictionary of options to use for the model. options - A dictionary of options to use for the model.
language - The language of the model. (use English for now) language - The language of the model. (use English for now)
model - The WhisperModel instance to use. model - The WhisperModel instance to use.

1
whisperx/transcribe.py
View File

@@ -119,7 +119,6 @@ def transcribe_task(args: dict, parser: argparse.ArgumentParser):
# Part 1: VAD & ASR Loop # Part 1: VAD & ASR Loop
results = [] results = []
tmp_results = []
# model = load_model(model_name, device=device, download_root=model_dir) # model = load_model(model_name, device=device, download_root=model_dir)
model = load_model( model = load_model(
model_name, model_name,