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