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

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

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

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

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