move model to assets

This commit is contained in:
Max Bain
2024-12-14 22:53:53 -06:00
parent a83ddbdf9b
commit a9e50ef0af
2 changed files with 1 additions and 1 deletions

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)
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