Remove warnings and add some warnings (#1408)

2024-08-06 18:37:55 +08:00
parent 2196b2bfc6
commit 805dbd7ff2
5 changed files with 61 additions and 10 deletions
--- a/GPT_SoVITS/prepare_datasets/1-get-text.py
+++ b/GPT_SoVITS/prepare_datasets/1-get-text.py
@@ -54,6 +54,8 @@ if os.path.exists(txt_path) == False:
    #     device = "mps"
    else:
        device = "cpu"
+    if os.path.exists(bert_pretrained_dir):...
+    else:raise FileNotFoundError(bert_pretrained_dir)
    tokenizer = AutoTokenizer.from_pretrained(bert_pretrained_dir)
    bert_model = AutoModelForMaskedLM.from_pretrained(bert_pretrained_dir)
    if is_half == True: