src/transformers/models/esm/modeling_tf_esm.py/TFEsmModel/__init__
class TFEsmModel:    def __init__(self, config: EsmConfig, add_pooling_layer=True, *inputs, **kwargs):
        super().__init__(config, *inputs, **kwargs)

        self.esm = TFEsmMainLayer(config, add_pooling_layer=add_pooling_layer, name="esm")