Hobis commited on
Commit
8f086d2
1 Parent(s): 0f60934

Update hubert/hubert_manager.py

Browse files
Files changed (1) hide show
  1. hubert/hubert_manager.py +1 -1
hubert/hubert_manager.py CHANGED
@@ -20,7 +20,7 @@ class HuBERTManager:
20
 
21
 
22
  @staticmethod
23
- def make_sure_tokenizer_installed(model: str = 'quantifier_hubert_base_ls960_14.pth', repo: str = 'GitMylo/bark-voice-cloning', local_file: str = 'tokenizer.pth'):
24
  install_dir = os.path.join('data', 'models', 'hubert')
25
  if not os.path.isdir(install_dir):
26
  os.makedirs(install_dir, exist_ok=True)
 
20
 
21
 
22
  @staticmethod
23
+ def make_sure_tokenizer_installed(model: str = 'polish-HuBERT-quantizer_8_epoch.pth', repo: str = 'Hobis/bark-voice-cloning-polish-HuBERT-quantizer', local_file: str = 'tokenizer.pth'):
24
  install_dir = os.path.join('data', 'models', 'hubert')
25
  if not os.path.isdir(install_dir):
26
  os.makedirs(install_dir, exist_ok=True)