src/transformers/models/squeezebert/modeling_squeezebert.py/SqueezeBertOnlyMLMHead/forward
class SqueezeBertOnlyMLMHead:    def forward(self, sequence_output):
        prediction_scores = self.predictions(sequence_output)
        return prediction_scores