def download_torch_align_models()

in whisperx-model-fetch/download_whisperx_models.py [0:0]


def download_torch_align_models():
    for lang, model_name in DEFAULT_ALIGN_MODELS_TORCH.items():
        print(f"Downloading {model_name} for {lang}")
        bundle = torchaudio.pipelines.__dict__[model_name]
        bundle.get_model()
        print(f"Downloaded {model_name} for {lang}")