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

Fix raise

Browse Source
This commit is contained in:
enhuiz 2023-01-13 00:34:40 +08:00
parent 36e8894f5c
commit f962a9d744
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vall_e/vall_e/base.py
View File

@ -165,6 +165,8 @@ class PrenormResidual(nn.Module):
elif norm_type == "adaln": elif norm_type == "adaln":
assert n_levels is not None assert n_levels is not None
self.norm = AdaLN(d_model, n_levels) self.norm = AdaLN(d_model, n_levels)
else:
raise NotImplementedError(norm_type)
self.dropout = nn.Dropout(p_dropout) self.dropout = nn.Dropout(p_dropout)
def forward(self, x, m, l): def forward(self, x, m, l):