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

oops

Browse Source
This commit is contained in:
mrq 2025-02-23 12:09:56 -06:00
parent b39aaacd77
commit 8f3c3e01ee
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vall_e/models/base.py
View File

@ -736,7 +736,7 @@ class Base(nn.Module):
self.audio_decoder = AudioDecoder( self.audio_decoder = AudioDecoder(
d_model, d_model,
d_model * 2, d_model * 2,
(n_audio_tokens + 2) * self.n_resp_levels, (n_audio_tokens + 1) * self.n_resp_levels,
) )
if attention_backend == "auto": if attention_backend == "auto":