src/transformers/models/lxmert/modeling_tf_lxmert.py/TFLxmertLMPredictionHead/set_output_embeddings
class TFLxmertLMPredictionHead:    def set_output_embeddings(self, value: tf.Variable):
        self.input_embeddings.weight = value
        self.input_embeddings.vocab_size = shape_list(value)[0]