Spaces:
Running
Running
Update ljspeechimportable.py
Browse files- 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 =
|
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)
|