From 19c55eece13a9f667a89528a89c86bb9ce2036d7 Mon Sep 17 00:00:00 2001 From: zhouzaida Date: Fri, 21 May 2021 14:15:06 +0800 Subject: [PATCH] [Fix] Remove hash validation temporarily when downloading checkpoint --- mim/commands/download.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/mim/commands/download.py b/mim/commands/download.py index 8ff31ff..f40456a 100644 --- a/mim/commands/download.py +++ b/mim/commands/download.py @@ -85,13 +85,11 @@ def download(package: str, checkpoint_urls = model_info[config]['weight'] for checkpoint_url in checkpoint_urls.split(','): filename = checkpoint_url.split('/')[-1] - hash_prefix = filename.split('.')[0].split('-')[-1] checkpoint_path = osp.join(dest_root, filename) if osp.exists(checkpoint_path): echo_success(f'{filename} exists in {dest_root}') else: - download_from_file(checkpoint_url, checkpoint_path, - hash_prefix) + download_from_file(checkpoint_url, checkpoint_path) echo_success( f'Successfully downloaded {filename} to {dest_root}')