src/transformers/models/esm/modeling_tf_esm.py/TFEsmForTokenClassification/__init__
class TFEsmForTokenClassification:    def __init__(self, config):
        super().__init__(config)
        self.num_labels = config.num_labels

        self.esm = TFEsmMainLayer(config, add_pooling_layer=False, name="esm")
        self.dropout = Dropout(config.hidden_dropout_prob)
        self.classifier = Dense(config.num_labels, name="classifier")