src/transformers/models/opt/modeling_tf_opt.py/TFOPTDecoder/set_input_embeddings
class TFOPTDecoder:    def set_input_embeddings(self, new_embeddings):
        self.embed_tokens.vocab_size = new_embeddings.shape[0]
        self.embed_tokens.weight = new_embeddings