src/transformers/models/funnel/modeling_funnel.py/FunnelForMaskedLM/set_output_embeddings
class FunnelForMaskedLM:    def set_output_embeddings(self, new_embeddings: nn.Embedding) -> None:
        self.lm_head = new_embeddings