whisperx.load_model

[wwfcnu]

whisperx.load_model

[GroovyDan]

Yes just pass the local file path as the whisper_arch argument:

print(">> Loading VAD Model")
vad_model = whisperx.vad.load_vad_model(
    torch.device(DEVICE),
    model_fp=os.path.join(MODEL_DIR, VAD_FOLDER, VAD_FILE),
)
print(">> Loading Whisper Model")
asr_model = whisperx.load_model(
    os.path.join(MODEL_DIR, ASR_FOLDER),
    DEVICE,
    compute_type=COMPUTE_TYPE,
    language="en",
    vad_model=vad_model,
)