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

mkdir for torch cache in case it doesnt exist

Browse Source
This commit is contained in:
m-bain
2023-04-01 13:05:40 -07:00
committed by GitHub
parent 861379edc3
commit a582a59493
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
whisperx/vad.py
View File

@ -18,6 +18,7 @@ VAD_SEGMENTATION_URL = "https://whisperx.s3.eu-west-2.amazonaws.com/model_weight
def load_vad_model(device, vad_onset, vad_offset, use_auth_token=None):
model_dir = torch.hub._get_torch_home()
os.makedirs(model_dir, exist_ok = True)
model_fp = os.path.join(model_dir, "whisperx-vad-segmentation.bin")
if os.path.exists(model_fp) and not os.path.isfile(model_fp):
raise RuntimeError(f"{model_fp} exists and is not a regular file")
@ -301,4 +302,4 @@ def merge_chunks(segments, chunk_size):
"end": curr_end,
"segments": seg_idxs,
})
return merged_segments
return merged_segments