From 5b56b54da26d03b1bf3578cdbaaee4acfe5e9e67 Mon Sep 17 00:00:00 2001 From: Marko Henning Date: Tue, 10 Sep 2024 09:37:46 +0200 Subject: [PATCH] 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")