src/transformers/models/esm/modeling_tf_esm.py/TFEsmModel/predict_contacts
class TFEsmModel:    def predict_contacts(self, tokens, attention_mask):
        return self.esm.predict_contacts(tokens, attention_mask)