heimoshuiyu/whisper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

infer download_root from XDG_CACHE_HOME if avail (#257)

Browse Source
This commit is contained in:
David Marx
2022-10-09 02:14:03 -07:00
committed by GitHub
parent 35713c66e0
commit 82725cea9c
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
whisper/__init__.py
View File

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