mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

ugh

Browse Source
This commit is contained in:
mrq 2025-03-25 23:24:01 -05:00
parent 8641c87611
commit 09e9438941
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
vall_e/models/base_v2.py
View File

@ -1219,7 +1219,6 @@ class Base_V2(nn.Module):
if tasks[0] == "len": if tasks[0] == "len":
# do duration prediction # do duration prediction
logits_aux = self.len_decoder( output.logits ) logits_aux = self.len_decoder( output.logits )
print( logits_aux[0].shape, logits_aux[0] )
# it's more accurate this way # it's more accurate this way
logits_aux = [ logit[..., -1, :1] for logit, aux_len in zip(logits_aux, aux_lens) ] logits_aux = [ logit[..., -1, :1] for logit, aux_len in zip(logits_aux, aux_lens) ]