Manmay commited on
Commit
0054bb4
1 Parent(s): 840a99d

change urls

Browse files
Files changed (1) hide show
  1. tortoise/api.py +5 -5
tortoise/api.py CHANGED
@@ -25,17 +25,17 @@ from tortoise.utils.tokenizer import VoiceBpeTokenizer
25
  from tortoise.utils.wav2vec_alignment import Wav2VecAlignment
26
  from contextlib import contextmanager
27
  from tortoise.models.stream_generator import init_stream_support
28
- # from huggingface_hub import hf_hub_download
29
  pbar = None
30
  init_stream_support()
31
  DEFAULT_MODELS_DIR = os.path.join(os.path.expanduser('~'), '.cache', 'tortoise', 'models')
32
  MODELS_DIR = os.environ.get('TORTOISE_MODELS_DIR', DEFAULT_MODELS_DIR)
33
 
34
  MODELS = {
35
- 'autoregressive.pth': 'https://huggingface.co/Manmay/tortoise-tts/blob/main/autoregressive.pth',
36
- 'classifier.pth': 'https://huggingface.co/Manmay/tortoise-tts/blob/main/classifier.pth',
37
- 'rlg_auto.pth': 'https://huggingface.co/Manmay/tortoise-tts/blob/main/rlg_auto.pth',
38
- 'hifidecoder.pth': 'https://huggingface.co/Manmay/tortoise-tts/blob/main/hifidecoder.pth',
39
  }
40
 
41
  def download_models(specific_models=None):
 
25
  from tortoise.utils.wav2vec_alignment import Wav2VecAlignment
26
  from contextlib import contextmanager
27
  from tortoise.models.stream_generator import init_stream_support
28
+ from huggingface_hub import hf_hub_download
29
  pbar = None
30
  init_stream_support()
31
  DEFAULT_MODELS_DIR = os.path.join(os.path.expanduser('~'), '.cache', 'tortoise', 'models')
32
  MODELS_DIR = os.environ.get('TORTOISE_MODELS_DIR', DEFAULT_MODELS_DIR)
33
 
34
  MODELS = {
35
+ 'autoregressive.pth': 'https://huggingface.co/Manmay/tortoise-tts/resolve/main/autoregressive.pth',
36
+ 'classifier.pth': 'https://huggingface.co/Manmay/tortoise-tts/resolve/main/classifier.pth',
37
+ 'rlg_auto.pth': 'https://huggingface.co/Manmay/tortoise-tts/resolve/main/rlg_auto.pth',
38
+ 'hifidecoder.pth': 'https://huggingface.co/Manmay/tortoise-tts/resolve/main/hifidecoder.pth',
39
  }
40
 
41
  def download_models(specific_models=None):