Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-21 21:58:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

test weights: check hash of pre-downloaded weights

Browse source
This commit is contained in:
Cédric Deltheil 2024-03-26 14:39:12 +00:00 committed by Cédric Deltheil
parent 04daeced73
commit 2345f01dd3
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
scripts/prepare_test_weights.py
View file

@ -96,8 +96,10 @@ def download_file(
print(f"❌{skip_icon} {response.status_code} ERROR {readable_size:<8} {url}") print(f"❌{skip_icon} {response.status_code} ERROR {readable_size:<8} {url}")
return return
if skip_existing and os.path.exists(dest_filename): if skip_existing and is_downloaded:
print(f"{skip_icon}️ Skipping previously downloaded {url}") print(f"{skip_icon}️ Skipping previously downloaded {url}")
if expected_hash is not None:
check_hash(dest_filename, expected_hash)
return return
os.makedirs(dest_folder, exist_ok=True) os.makedirs(dest_folder, exist_ok=True)