Respair commited on
Commit
02e714f
1 Parent(s): f490eec

Update ljspeechimportable.py

Browse files
Files changed (1) hide show
  1. ljspeechimportable.py +1 -1
ljspeechimportable.py CHANGED
@@ -650,7 +650,7 @@ pitch_extractor = load_F0_models(F0_path)
650
 
651
  # load BERT model
652
  from Utils.PLBERT.util import load_plbert
653
- BERT_path = "/home/user/app/Utils/PLBERT/step_1040000.t7"
654
  plbert = load_plbert(BERT_path)
655
 
656
  model = build_model(recursive_munch(config['model_params']), text_aligner, pitch_extractor, plbert)
 
650
 
651
  # load BERT model
652
  from Utils.PLBERT.util import load_plbert
653
+ BERT_path = config.get('PLBERT_dir', False)
654
  plbert = load_plbert(BERT_path)
655
 
656
  model = build_model(recursive_munch(config['model_params']), text_aligner, pitch_extractor, plbert)