blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-06 03:57:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

name2go fix for fast_inference (#1133)

Browse Source
This commit is contained in:
XXXXRT666 2024-05-26 17:45:20 +01:00 committed by GitHub
parent f822b9588f
commit 8fc1e34f58
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
GPT_SoVITS/prepare_datasets/2-get-hubert-wav32k.py
View File

@ -82,7 +82,7 @@ def name2go(wav_name,wav_path):
tensor_wav16 = tensor_wav16.to(device) tensor_wav16 = tensor_wav16.to(device)
ssl=model.model(tensor_wav16.unsqueeze(0))["last_hidden_state"].transpose(1,2).cpu()#torch.Size([1, 768, 215]) ssl=model.model(tensor_wav16.unsqueeze(0))["last_hidden_state"].transpose(1,2).cpu()#torch.Size([1, 768, 215])
if np.isnan(ssl.detach().numpy()).sum()!= 0: if np.isnan(ssl.detach().numpy()).sum()!= 0:
nan_fails.append(wav_name) nan_fails.append((wav_name,wav_path))
print("nan filtered:%s"%wav_name) print("nan filtered:%s"%wav_name)
return return
wavfile.write( wavfile.write(
@ -90,7 +90,7 @@ def name2go(wav_name,wav_path):
32000, 32000,
tmp_audio32.astype("int16"), tmp_audio32.astype("int16"),
) )
my_save(ssl,hubert_path ) my_save(ssl,hubert_path)
with open(inp_text,"r",encoding="utf8")as f: with open(inp_text,"r",encoding="utf8")as f:
lines=f.read().strip("\n").split("\n") lines=f.read().strip("\n").split("\n")
@ -113,8 +113,8 @@ for line in lines[int(i_part)::int(all_parts)]:
if(len(nan_fails)>0 and is_half==True): if(len(nan_fails)>0 and is_half==True):
is_half=False is_half=False
model=model.float() model=model.float()
for wav_name in nan_fails: for wav in nan_fails:
try: try:
name2go(wav_name) name2go(wav[0],wav[1])
except: except:
print(wav_name,traceback.format_exc()) print(wav_name,traceback.format_exc())