Fixed pyannote_cache, now looks it up before overwrite.

2024-09-10 13:50:28 +02:00
parent 5b56b54da2
commit 885d0c864e
1 changed files with 4 additions and 1 deletions
@@ -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")