src/transformers/models/switch_transformers/modeling_switch_transformers.py/SwitchTransformersForConditionalGeneration/get_output_embeddings
class SwitchTransformersForConditionalGeneration:    def get_output_embeddings(self):
        return self.lm_head