Manmay commited on
Commit
fa68c71
1 Parent(s): 78648f9
Files changed (1) hide show
  1. 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
- # DEFAULT_MODELS_DIR = os.path.join(os.path.expanduser('~'), '.cache', 'tortoise', 'models')
29
- MODELS_DIR = "/data/speech_synth/models"
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',