src/transformers/models/flaubert/modeling_flaubert.py/TransformerFFN/ff_chunk
class TransformerFFN:    def ff_chunk(self, input):
        x = self.lin1(input)
        x = self.act(x)
        x = self.lin2(x)
        x = nn.functional.dropout(x, p=self.dropout, training=self.training)
        return x