src/transformers/models/esm/modeling_esm.py/EsmForMaskedLM/predict_contacts
class EsmForMaskedLM:    def predict_contacts(self, tokens, attention_mask):
        return self.esm.predict_contacts(tokens, attention_mask=attention_mask)