Respair commited on
Commit
f490eec
1 Parent(s): 9b89c57

Update styletts2importable.py

Browse files
Files changed (1) hide show
  1. styletts2importable.py +1 -1
styletts2importable.py CHANGED
@@ -726,7 +726,7 @@ pitch_extractor = load_F0_models(F0_path)
726
 
727
  # load BERT model
728
  from Utils.PLBERT.util import load_plbert
729
- BERT_path = "/home/user/app/Utils/PLBERT/step_1040000.t7"
730
  plbert = load_plbert(BERT_path)
731
 
732
  model_params = recursive_munch(config['model_params'])
 
726
 
727
  # load BERT model
728
  from Utils.PLBERT.util import load_plbert
729
+ BERT_path = config.get('PLBERT_dir', False)
730
  plbert = load_plbert(BERT_path)
731
 
732
  model_params = recursive_munch(config['model_params'])