src/transformers/models/funnel/modeling_tf_funnel.py/TFFunnelMaskedLMHead/__init__
class TFFunnelMaskedLMHead:    def __init__(self, config, input_embeddings, **kwargs):
        super().__init__(**kwargs)
        self.config = config
        self.hidden_size = config.hidden_size
        self.input_embeddings = input_embeddings