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