add unit tests.

Signed-off-by: Xuesong Yang <[email protected]>
NVIDIA · Jan 18, 2025 · d6b05ac · d6b05ac
1 parent 1aad73e
commit d6b05ac
Show file tree

Hide file tree

Showing 2 changed files with 431 additions and 0 deletions.
diff --git a/nemo/collections/tts/modules/transformer_2412.py b/nemo/collections/tts/modules/transformer_2412.py
@@ -576,6 +576,7 @@ def __init__(
         """
         if has_xattn and (xa_d_memory is None or xa_n_heads is None):
             raise ValueError("It requires that `xa_d_memory` and `xa_n_heads` are specified when `has_xattn` is True!")
+
         super().__init__()
         self.dropout = torch.nn.Dropout(p_dropout)
         self.p_dropout_out = p_dropout_out