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 #945 from m-bain/m-bain/local_model

Browse Source 
move model to assets
This commit is contained in:
Max Bain
2024-12-14 22:54:56 -06:00
committed by GitHub
parent 161ae1f7ad a9e50ef0af
commit c141074cbd
2 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
models/pytorch_model.bin → whisperx/assets/pytorch_model.bin
View File

2
whisperx/vad.py
View File

@ -26,7 +26,7 @@ def load_vad_model(device, vad_onset=0.500, vad_offset=0.363, use_auth_token=Non
os.makedirs(model_dir, exist_ok = True) os.makedirs(model_dir, exist_ok = True)
if model_fp is None: if model_fp is None:
# Dynamically resolve the path to the model file # Dynamically resolve the path to the model file
model_fp = os.path.join(vad_dir, "..", "models", "pytorch_model.bin") model_fp = os.path.join(vad_dir, "assets", "pytorch_model.bin")
model_fp = os.path.abspath(model_fp) # Ensure the path is absolute model_fp = os.path.abspath(model_fp) # Ensure the path is absolute
else: else:
model_fp = os.path.abspath(model_fp) # Ensure any provided path is absolute model_fp = os.path.abspath(model_fp) # Ensure any provided path is absolute