From 885d0c864ebd33f75416dd3300eca88f79c527c7 Mon Sep 17 00:00:00 2001 From: Marko Henning Date: Tue, 10 Sep 2024 13:50:28 +0200 Subject: [PATCH] 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 @@ "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")