Fix raise

2023-01-13 00:34:40 +08:00 · 2023-01-13 00:34:40 +08:00 · f962a9d744
commit f962a9d744
parent 36e8894f5c
1 changed files with 2 additions and 0 deletions
--- a/vall_e/vall_e/base.py
+++ b/vall_e/vall_e/base.py
@ -165,6 +165,8 @@ class PrenormResidual(nn.Module):
        elif norm_type == "adaln":
            assert n_levels is not None
            self.norm = AdaLN(d_model, n_levels)
+        else:
+            raise NotImplementedError(norm_type)
        self.dropout = nn.Dropout(p_dropout)

    def forward(self, x, m, l):