admin/scribe
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Merge pull request #120 from JSchmie/fix_env_var

Browse Source 
Fixed pyannote env var import, now can use `PYANNOTE_CACHE`
This commit is contained in:
Jacob Schmieder
2024-09-10 15:24:44 +02:00
committed by GitHub
parent 0ddb52cc95 885d0c864e
commit 4e7b7e748b
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scraibe/misc.py
+4 -4
View File
@@ -1,6 +1,5 @@
import os import os
import yaml import yaml
from pyannote.audio.core.model import CACHE_DIR as PYANNOTE_CACHE_DIR
from argparse import Action from argparse import Action
from ast import literal_eval from ast import literal_eval
@@ -8,9 +7,10 @@ CACHE_DIR = os.getenv(
"AUTOT_CACHE", "AUTOT_CACHE",
os.path.expanduser("~/.cache/torch/models"), os.path.expanduser("~/.cache/torch/models"),
) )
os.getenv(
if CACHE_DIR != PYANNOTE_CACHE_DIR: "PYANNOTE_CACHE",
os.environ["PYANNOTE_CACHE"] = os.path.join(CACHE_DIR, "pyannote") os.path.join(CACHE_DIR, "pyannote"),
)
WHISPER_DEFAULT_PATH = os.path.join(CACHE_DIR, "whisper") WHISPER_DEFAULT_PATH = os.path.join(CACHE_DIR, "whisper")
PYANNOTE_DEFAULT_PATH = os.path.join(CACHE_DIR, "pyannote") PYANNOTE_DEFAULT_PATH = os.path.join(CACHE_DIR, "pyannote")