mirror of
https://github.com/openai/whisper.git
synced 2025-11-24 06:26:03 +00:00
Update __init__.py
Using "with" to close file immediately after reading instead of leaving it open for garbage collector
This commit is contained in:
parent
c0d2f624c0
commit
de060c7d17
@ -86,7 +86,8 @@ def _download(url: str, root: str, in_memory: bool) -> Union[bytes, str]:
|
|||||||
output.write(buffer)
|
output.write(buffer)
|
||||||
loop.update(len(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:
|
if hashlib.sha256(model_bytes).hexdigest() != expected_sha256:
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"Model has been downloaded but the SHA256 checksum does not not match. Please retry loading the model."
|
"Model has been downloaded but the SHA256 checksum does not not match. Please retry loading the model."
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user