blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-07 15:19:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update utils.py

Browse Source 
make it half half
This commit is contained in:
XXXXRT666 2024-05-02 01:51:27 +01:00 committed by GitHub
parent d2f991bb6d
commit b7d822df93
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
GPT_SoVITS/AR/models/utils.py
View File

@ -208,16 +208,17 @@ def make_reject_y(y_o, y_lens):
bs = len(y_lens) bs = len(y_lens)
reject_y = [] reject_y = []
reject_y_lens = [] reject_y_lens = []
process_item_idx = True
for b in range(bs): for b in range(bs):
process_item_idx = torch.randint(0, 2, size=(1, ))[0] if process_item_idx == True:
if process_item_idx == 0:
new_y = repeat_P(y_o[b]) new_y = repeat_P(y_o[b])
reject_y.append(new_y) reject_y.append(new_y)
reject_y_lens.append(len(new_y)) reject_y_lens.append(len(new_y))
elif process_item_idx==1: elif process_item_idx== False:
new_y = lost_P(y_o[b]) new_y = lost_P(y_o[b])
reject_y.append(new_y) reject_y.append(new_y)
reject_y_lens.append(len(new_y)) reject_y_lens.append(len(new_y))
process_item_idx = not process_item_idx
max_length = max(reject_y_lens) max_length = max(reject_y_lens)
for b in range(bs): for b in range(bs):
pad_length = max_length - reject_y_lens[b] pad_length = max_length - reject_y_lens[b]