src/transformers/models/imagegpt/modeling_imagegpt.py/ImageGPTForCausalImageModeling/get_output_embeddings
class ImageGPTForCausalImageModeling:    def get_output_embeddings(self):
        return self.lm_head