File size: 156 Bytes
ed95978
 
 
 
1
2
3
4
5
def download_model(tag: str, out_dir: str):
    from huggingface_hub import snapshot_download

    return snapshot_download(repo_id=tag, local_dir=out_dir)