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

Handle XDG_CACHE_HOME properly for download_root (#864)

Browse Source 
Co-authored-by: Jong Wook Kim <jongwook@openai.com>
This commit is contained in:
zer0-x
2023-01-21 12:09:39 +03:00
committed by GitHub
parent 12e1089462
commit 9f7aba6099
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
whisper/__init__.py
View File

@@ -94,9 +94,14 @@ 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.getenv( download_root = os.path.join(
"XDG_CACHE_HOME", os.getenv(
os.path.join(os.path.expanduser("~"), ".cache", "whisper") "XDG_CACHE_HOME",
os.path.join(
os.path.expanduser("~"), ".cache"
)
),
"whisper"
) )
if name in _MODELS: if name in _MODELS: