Update README.md

README.md

@@ -31,9 +31,12 @@ Notable differences compared to the default PL-BERT checkpoint and config availa
 
 ## How do I train StyleTTS2 with this new PL-BERT checkpoint?
 
-* Simply create a new folder under `Utils` in your StyleTTS2 repository. Call it, for example, `PLBERT-all-languages`. 
+* Simply create a new folder under `Utils` in your StyleTTS2 repository. Call it, for example, `PLBERT_all_languages`. 
 * Copy paste into it `config.yml`, `step_1100000.t7` and `util.py`.
-* Then, in your StyleTTS2 config file, change `PLBERT_dir` to `Utils/PLBERT-all-languages`.
+* Then, in your StyleTTS2 config file, change `PLBERT_dir` to `Utils/PLBERT_all_languages`. You will also need to change your import as such:
+  * Change `from Utils.PLBERT.util import load_plbert`
+  * To `from Utils.PLBERT_all_languages.util import load_plbert`
+  * Alternatively, you can just replace the relevant files in `Utils/PLBERT` and not have to change any code.
 * Now, you need to create train and validation files. You will need to use `espeak` to create a file in the same format as the ones that exist in the `Data` folder of the StyleTTS2 repository. Careful! You will need to change the `language` argument to phonemise your text if it's not in English. You can find the correct language codes [here](https://github.com/espeak-ng/espeak-ng/blob/master/docs/languages.md). For example, Latin American Spanish is `es-419`
 
 Voila, you can now train a multilingual StyleTTS2 model!