From 0e2467ace44e2b4d9053398d8be84c4d49da8bd6 Mon Sep 17 00:00:00 2001
From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com>
Date: Tue, 16 Jan 2024 22:25:15 +0800
Subject: [PATCH] Update 1-get-text.py

---
 GPT_SoVITS/prepare_datasets/1-get-text.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/GPT_SoVITS/prepare_datasets/1-get-text.py b/GPT_SoVITS/prepare_datasets/1-get-text.py
index 4f2d95a..5abd353 100644
--- a/GPT_SoVITS/prepare_datasets/1-get-text.py
+++ b/GPT_SoVITS/prepare_datasets/1-get-text.py
@@ -90,7 +90,15 @@ if(os.path.exists(txt_path)==False):
         lines=f.read().strip("\n").split("\n")
 
     language_v1_to_language_v2={
-        "ZH":"zh"
+        "ZH":"zh",
+        "zh":"zh",
+        "JP":"ja",
+        "jp":"ja",
+        "JA":"ja",
+        "ja":"ja",
+        "EN":"en",
+        "en":"en",
+        "En":"en",
     }
     for line in lines[int(i_part)::int(all_parts)]:
         try: