from local

This commit is contained in:
dohe0342 2023-02-14 17:54:31 +09:00
parent 01fbc6c4b9
commit bce8655058
2 changed files with 1 additions and 0 deletions

View File

@ -405,6 +405,7 @@ class ConformerEncoder(nn.Module):
mask: Optional[torch.Tensor] = None,
src_key_padding_mask: Optional[torch.Tensor] = None,
warmup: float = 1.0,
condition_layer = None,
) -> torch.Tensor:
"""
Pass the input through the encoder layers in turn.