blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-05 19:41:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix indentation issue

Browse Source
This commit is contained in:
Wu Zichen 2024-01-26 01:40:18 +08:00
parent 7259cc95fd
commit 3029560356
2 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
GPT_SoVITS/prepare_datasets/1-get-text.py
View File

@ -47,12 +47,12 @@ if os.path.exists(txt_path) == False:
bert_dir = "%s/3-bert" % (opt_dir) bert_dir = "%s/3-bert" % (opt_dir)
os.makedirs(opt_dir, exist_ok=True) os.makedirs(opt_dir, exist_ok=True)
os.makedirs(bert_dir, exist_ok=True) os.makedirs(bert_dir, exist_ok=True)
if torch.cuda.is_available(): if torch.cuda.is_available():
device = "cuda:0" device = "cuda:0"
elif torch.backends.mps.is_available(): elif torch.backends.mps.is_available():
device = "mps" device = "mps"
else: else:
device = "cpu" device = "cpu"
tokenizer = AutoTokenizer.from_pretrained(bert_pretrained_dir) tokenizer = AutoTokenizer.from_pretrained(bert_pretrained_dir)
bert_model = AutoModelForMaskedLM.from_pretrained(bert_pretrained_dir) bert_model = AutoModelForMaskedLM.from_pretrained(bert_pretrained_dir)
if is_half == True: if is_half == True:

12
GPT_SoVITS/prepare_datasets/3-get-semantic.py
View File

@ -38,12 +38,12 @@ semantic_path = "%s/6-name2semantic-%s.tsv" % (opt_dir, i_part)
if os.path.exists(semantic_path) == False: if os.path.exists(semantic_path) == False:
os.makedirs(opt_dir, exist_ok=True) os.makedirs(opt_dir, exist_ok=True)
if torch.cuda.is_available(): if torch.cuda.is_available():
device = "cuda" device = "cuda"
elif torch.backends.mps.is_available(): elif torch.backends.mps.is_available():
device = "mps" device = "mps"
else: else:
device = "cpu" device = "cpu"
hps = utils.get_hparams_from_file(s2config_path) hps = utils.get_hparams_from_file(s2config_path)
vq_model = SynthesizerTrn( vq_model = SynthesizerTrn(
hps.data.filter_length // 2 + 1, hps.data.filter_length // 2 + 1,