src/transformers/models/ctrl/modeling_tf_ctrl.py/TFCTRLMainLayer/set_input_embeddings
class TFCTRLMainLayer:    def set_input_embeddings(self, value):
        self.w.weight = value
        self.w.vocab_size = shape_list(value)[0]