blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-07 15:19:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bugs

Browse Source
This commit is contained in:
XXXXRT666 2024-08-05 23:12:17 +08:00
parent 990e9f43f9
commit f13685575b
3 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
GPT_SoVITS/AR/data/dataset.py
View File

@ -15,6 +15,8 @@ from torch.utils.data import DataLoader
from torch.utils.data import Dataset from torch.utils.data import Dataset
from transformers import AutoTokenizer from transformers import AutoTokenizer
version = os.environ.get('version',None)
from text import cleaned_text_to_sequence from text import cleaned_text_to_sequence
# from config import exp_dir # from config import exp_dir
@ -149,7 +151,7 @@ class Text2SemanticDataset(Dataset):
phoneme = phoneme.split(" ") phoneme = phoneme.split(" ")
try: try:
phoneme_ids = cleaned_text_to_sequence(phoneme) phoneme_ids = cleaned_text_to_sequence(phoneme, version)
except: except:
traceback.print_exc() traceback.print_exc()
# print(f"{item_name} not in self.phoneme_data !") # print(f"{item_name} not in self.phoneme_data !")

4
GPT_SoVITS/module/data_utils.py
View File

@ -18,7 +18,7 @@ import requests
from scipy.io import wavfile from scipy.io import wavfile
from io import BytesIO from io import BytesIO
from tools.my_utils import load_audio from tools.my_utils import load_audio
version = os.environ.get('version',None)
# ZeroDivisionError fixed by Tybost (https://github.com/RVC-Boss/GPT-SoVITS/issues/79) # ZeroDivisionError fixed by Tybost (https://github.com/RVC-Boss/GPT-SoVITS/issues/79)
class TextAudioSpeakerLoader(torch.utils.data.Dataset): class TextAudioSpeakerLoader(torch.utils.data.Dataset):
""" """
@ -77,7 +77,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
try: try:
phoneme = self.phoneme_data[audiopath][0] phoneme = self.phoneme_data[audiopath][0]
phoneme = phoneme.split(' ') phoneme = phoneme.split(' ')
phoneme_ids = cleaned_text_to_sequence(phoneme) phoneme_ids = cleaned_text_to_sequence(phoneme, version)
except Exception: except Exception:
print(f"{audiopath} not in self.phoneme_data !") print(f"{audiopath} not in self.phoneme_data !")
skipped_phone += 1 skipped_phone += 1

6
GPT_SoVITS/text/cleaner.py
View File

@ -84,9 +84,11 @@ def clean_special(text, language, special_s, target_symbol, version=None):
return new_ph, phones[1], norm_text return new_ph, phones[1], norm_text
def text_to_sequence(text, language): def text_to_sequence(text, language, version=None):
version = os.environ.get('version',version)
if version is None:version='v2'
phones = clean_text(text) phones = clean_text(text)
return cleaned_text_to_sequence(phones) return cleaned_text_to_sequence(phones, version)
if __name__ == "__main__": if __name__ == "__main__":