From 5b56b54da26d03b1bf3578cdbaaee4acfe5e9e67 Mon Sep 17 00:00:00 2001 From: Marko Henning Date: Tue, 10 Sep 2024 09:37:46 +0200 Subject: [PATCH 1/2] Fixed pyannote env var import, now can use `PYANNOTE_CACHE` --- scraibe/misc.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/scraibe/misc.py b/scraibe/misc.py index f12335f..7b7cbf4 100644 --- a/scraibe/misc.py +++ b/scraibe/misc.py @@ -1,6 +1,5 @@ import os import yaml -from pyannote.audio.core.model import CACHE_DIR as PYANNOTE_CACHE_DIR from argparse import Action from ast import literal_eval @@ -8,9 +7,7 @@ CACHE_DIR = os.getenv( "AUTOT_CACHE", os.path.expanduser("~/.cache/torch/models"), ) - -if CACHE_DIR != PYANNOTE_CACHE_DIR: - os.environ["PYANNOTE_CACHE"] = os.path.join(CACHE_DIR, "pyannote") +os.environ["PYANNOTE_CACHE"] = os.path.join(CACHE_DIR, "pyannote") WHISPER_DEFAULT_PATH = os.path.join(CACHE_DIR, "whisper") PYANNOTE_DEFAULT_PATH = os.path.join(CACHE_DIR, "pyannote") From 885d0c864ebd33f75416dd3300eca88f79c527c7 Mon Sep 17 00:00:00 2001 From: Marko Henning Date: Tue, 10 Sep 2024 13:50:28 +0200 Subject: [PATCH 2/2] Fixed pyannote_cache, now looks it up before overwrite. --- scraibe/misc.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/scraibe/misc.py b/scraibe/misc.py index 7b7cbf4..21099fb 100644 --- a/scraibe/misc.py +++ b/scraibe/misc.py @@ -7,7 +7,10 @@ CACHE_DIR = os.getenv( "AUTOT_CACHE", os.path.expanduser("~/.cache/torch/models"), ) -os.environ["PYANNOTE_CACHE"] = os.path.join(CACHE_DIR, "pyannote") +os.getenv( + "PYANNOTE_CACHE", + os.path.join(CACHE_DIR, "pyannote"), +) WHISPER_DEFAULT_PATH = os.path.join(CACHE_DIR, "whisper") PYANNOTE_DEFAULT_PATH = os.path.join(CACHE_DIR, "pyannote")