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 2024-11-14 22:18:33 -06:00
parent 39096f8ff3
commit 2fbeacfe92
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vall_e/models/base.py
View File

@ -471,9 +471,9 @@ class Base(nn.Module):
resp_l_names = ['AR:0:0'] + [f'NAR:{i}:{i+1}' for i in range( self.n_resp_levels - 1 )]
classifier_l_tokens = l_tokens + [ n_text_tokens ]
classifier_l_names = resp_l_names + [ "len" ]
classifier_l_names = resp_l_names + [ "stt" ]
if "len" in self.capabilities and False:
if "len" in self.capabilities:
classifier_l_tokens += [ n_text_tokens ]
classifier_l_names += ["len"]