Ver Fonte

Explicitly closing model file after reading it (#630)

Paul Harter há 2 anos atrás
pai
commit
fd8f80c8b8
1 ficheiros alterados com 2 adições e 1 exclusões
  1. 2 1
      whisper/__init__.py

+ 2 - 1
whisper/__init__.py

@@ -39,7 +39,8 @@ def _download(url: str, root: str, in_memory: bool) -> Union[bytes, str]:
         raise RuntimeError(f"{download_target} exists and is not a regular file")
 
     if os.path.isfile(download_target):
-        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:
             return model_bytes if in_memory else download_target
         else: