Skip to content

Commit 8fc1e34

Browse files
authored
name2go fix for fast_inference (RVC-Boss#1133)
1 parent f822b95 commit 8fc1e34

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

GPT_SoVITS/prepare_datasets/2-get-hubert-wav32k.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -82,15 +82,15 @@ def name2go(wav_name,wav_path):
8282
tensor_wav16 = tensor_wav16.to(device)
8383
ssl=model.model(tensor_wav16.unsqueeze(0))["last_hidden_state"].transpose(1,2).cpu()#torch.Size([1, 768, 215])
8484
if np.isnan(ssl.detach().numpy()).sum()!= 0:
85-
nan_fails.append(wav_name)
85+
nan_fails.append((wav_name,wav_path))
8686
print("nan filtered:%s"%wav_name)
8787
return
8888
wavfile.write(
8989
"%s/%s"%(wav32dir,wav_name),
9090
32000,
9191
tmp_audio32.astype("int16"),
9292
)
93-
my_save(ssl,hubert_path )
93+
my_save(ssl,hubert_path)
9494

9595
with open(inp_text,"r",encoding="utf8")as f:
9696
lines=f.read().strip("\n").split("\n")
@@ -113,8 +113,8 @@ def name2go(wav_name,wav_path):
113113
if(len(nan_fails)>0 and is_half==True):
114114
is_half=False
115115
model=model.float()
116-
for wav_name in nan_fails:
116+
for wav in nan_fails:
117117
try:
118-
name2go(wav_name)
118+
name2go(wav[0],wav[1])
119119
except:
120120
print(wav_name,traceback.format_exc())

0 commit comments

Comments
 (0)