From a9e50ef0afb876ca1468258a4670b636257938ac Mon Sep 17 00:00:00 2001 From: Max Bain Date: Sat, 14 Dec 2024 22:53:53 -0600 Subject: [PATCH] move model to assets --- {models => whisperx/assets}/pytorch_model.bin | Bin whisperx/vad.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) rename {models => whisperx/assets}/pytorch_model.bin (100%) diff --git a/models/pytorch_model.bin b/whisperx/assets/pytorch_model.bin similarity index 100% rename from models/pytorch_model.bin rename to whisperx/assets/pytorch_model.bin diff --git a/whisperx/vad.py b/whisperx/vad.py index 7f9aae3..c2db2b6 100644 --- a/whisperx/vad.py +++ b/whisperx/vad.py @@ -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) if model_fp is None: # 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 else: model_fp = os.path.abspath(model_fp) # Ensure any provided path is absolute