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

Fixed pyannote env var import, now can use PYANNOTE_CACHE

Browse Source
This commit is contained in:
Marko Henning
2024-09-10 09:37:46 +02:00
parent 0ddb52cc95
commit 5b56b54da2
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scraibe/misc.py
+1 -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,7 @@ CACHE_DIR = os.getenv(
"AUTOT_CACHE", "AUTOT_CACHE",
os.path.expanduser("~/.cache/torch/models"), os.path.expanduser("~/.cache/torch/models"),
) )
os.environ["PYANNOTE_CACHE"] = os.path.join(CACHE_DIR, "pyannote")
if CACHE_DIR != PYANNOTE_CACHE_DIR:
os.environ["PYANNOTE_CACHE"] = 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")