diff --git a/download-model.py b/download-model.py index 25d9c804e7..65f4fa5283 100644 --- a/download-model.py +++ b/download-model.py @@ -236,8 +236,7 @@ def check_model_files(self, model, branch, links, sha256, output_folder): continue with open(output_folder / sha256[i][0], "rb") as f: - bytes = f.read() - file_hash = hashlib.sha256(bytes).hexdigest() + file_hash = hashlib.file_digest(f, "sha256").hexdigest() if file_hash != sha256[i][1]: print(f'Checksum failed: {sha256[i][0]} {sha256[i][1]}') validated = False