tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Fix dir_exits error

Browse Source
This commit is contained in:
litagin02
2024-01-12 13:43:29 +09:00
parent 1067b8ec60
commit a8e0619ad4
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
text/japanese_bert.py
View File

@@ -49,7 +49,7 @@ def get_bert_feature(
style_res = torch.cat(style_res["hidden_states"][-3:-2], -1)[0].cpu() style_res = torch.cat(style_res["hidden_states"][-3:-2], -1)[0].cpu()
style_res_mean = style_res.mean(0) style_res_mean = style_res.mean(0)
assert len(word2ph) == len(text) + 2 assert len(word2ph) == len(text) + 2, text
word2phone = word2ph word2phone = word2ph
phone_level_feature = [] phone_level_feature = []
for i in range(len(word2phone)): for i in range(len(word2phone)):

1
webui_train.py
View File

@@ -65,6 +65,7 @@ def initialize(model_name, batch_size, epochs, save_every_steps, bf16_run):
shutil.copytree( shutil.copytree(
src=model_path, src=model_path,
dst=os.path.join(dataset_path, "models_backup"), dst=os.path.join(dataset_path, "models_backup"),
dirs_exist_ok=True,
) )
shutil.rmtree(model_path) shutil.rmtree(model_path)
try: try: