From 85a57be266f33d9d0496fb3a00e1ce4082d2a697 Mon Sep 17 00:00:00 2001
From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com>
Date: Fri, 2 Aug 2024 15:44:57 +0800
Subject: [PATCH] support gpt-sovits v2

support gpt-sovits v2
---
 GPT_SoVITS/prepare_datasets/1-get-text.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/GPT_SoVITS/prepare_datasets/1-get-text.py b/GPT_SoVITS/prepare_datasets/1-get-text.py
index 7be2508..20a4564 100644
--- a/GPT_SoVITS/prepare_datasets/1-get-text.py
+++ b/GPT_SoVITS/prepare_datasets/1-get-text.py
@@ -82,6 +82,7 @@ if os.path.exists(txt_path) == False:
         for name, text, lan in data:
             try:
                 name = os.path.basename(name)
+                print(name)
                 phones, word2ph, norm_text = clean_text(
                     text.replace("%", "-").replace("￥", ","), lan
                 )
@@ -112,6 +113,12 @@ if os.path.exists(txt_path) == False:
         "EN": "en",
         "en": "en",
         "En": "en",
+        "KO": "ko",
+        "Ko": "ko",
+        "ko": "ko",
+        "yue": "yue",
+        "YUE": "yue",
+        "Yue": "yue",
     }
     for line in lines[int(i_part) :: int(all_parts)]:
         try: