Handle XDG_CACHE_HOME properly for download_root (openai#864)

Co-authored-by: Jong Wook Kim <[email protected]>
ilanit1997 · May 16, 2023 · 2990115 · 2990115
1 parent 91d56d2
commit 2990115
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/whisper/__init__.py b/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: