Files
whisperbox-transcribe/scripts/download_models.py
2023-03-01 14:35:52 +01:00

10 lines
295 B
Python

import os
import sys
from whisper import _download, _MODELS # type: ignore
if __name__ == "__main__":
model_name = sys.argv[1].strip()
_download(_MODELS[model_name], "/models/", False)
# if model_name != "large":
# _download(_MODELS[f"{model_name}.en"], "/models/", False)