Spaces:
Running
on
T4
Running
on
T4
fix path
Browse files- tortoise/api.py +2 -2
tortoise/api.py
CHANGED
@@ -25,8 +25,8 @@ from tortoise.utils.wav2vec_alignment import Wav2VecAlignment
|
|
25 |
from contextlib import contextmanager
|
26 |
pbar = None
|
27 |
|
28 |
-
|
29 |
-
MODELS_DIR =
|
30 |
MODELS = {
|
31 |
'autoregressive.pth': 'https://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/autoregressive.pth',
|
32 |
'classifier.pth': 'https://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/classifier.pth',
|
|
|
25 |
from contextlib import contextmanager
|
26 |
pbar = None
|
27 |
|
28 |
+
DEFAULT_MODELS_DIR = os.path.join(os.path.expanduser('~'), '.cache', 'tortoise', 'models')
|
29 |
+
MODELS_DIR = os.environ.get('TORTOISE_MODELS_DIR', DEFAULT_MODELS_DIR)
|
30 |
MODELS = {
|
31 |
'autoregressive.pth': 'https://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/autoregressive.pth',
|
32 |
'classifier.pth': 'https://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/classifier.pth',
|