diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/models.py b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/models.py index 90a12b779..502cb093e 100644 --- a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/models.py +++ b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/models.py @@ -208,7 +208,7 @@ def __init__(self, config: ConformerConfig): config.feed_forward_residual_dropout_rate) self.dropout2 = nn.Dropout( p=feed_forward_residual_dropout_rate, inplace=True) - + def forward(self, inputs, padding_mask): inputs = self.ln(inputs) inputs = self.linear1(inputs)