Spaces:
Runtime error
Runtime error
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'])
|