Procházet zdrojové kódy

Explicitly closing model file after reading it (#630)

Paul Harter před 2 roky
rodič
revize
fd8f80c8b8
1 změnil soubory, kde provedl 2 přidání a 1 odebrání
  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: