Update __init__.py

Using "with" to close file immediately after reading instead of leaving it open for garbage collector
This commit is contained in:
Carson Rochon 2025-09-23 20:51:40 -06:00 committed by GitHub
parent c0d2f624c0
commit de060c7d17
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -86,7 +86,8 @@ def _download(url: str, root: str, in_memory: bool) -> Union[bytes, str]:
output.write(buffer)
loop.update(len(buffer))
model_bytes = open(download_target, "rb").read()
with open(download_target, "rb") as f:
model_bytes = f.read()
if hashlib.sha256(model_bytes).hexdigest() != expected_sha256:
raise RuntimeError(
"Model has been downloaded but the SHA256 checksum does not not match. Please retry loading the model."