Browse Source

Handle XDG_CACHE_HOME properly for download_root (#864)

Co-authored-by: Jong Wook Kim <jongwook@openai.com>
zer0-x 2 years ago
parent
commit
9f7aba6099
1 changed files with 8 additions and 3 deletions
  1. 8 3
      whisper/__init__.py

+ 8 - 3
whisper/__init__.py

@@ -94,9 +94,14 @@ def load_model(name: str, device: Optional[Union[str, torch.device]] = None, dow
     if device is None:
         device = "cuda" if torch.cuda.is_available() else "cpu"
     if download_root is None:
-        download_root = os.getenv(
-            "XDG_CACHE_HOME", 
-            os.path.join(os.path.expanduser("~"), ".cache", "whisper")
+        download_root = os.path.join(
+            os.getenv(
+                "XDG_CACHE_HOME",
+                os.path.join(
+                    os.path.expanduser("~"), ".cache"
+                )
+            ),
+            "whisper"
         )
 
     if name in _MODELS: