examples/research_projects/deebert/src/modeling_highway_bert.py/DeeBertModel/set_input_embeddings
class DeeBertModel:    def set_input_embeddings(self, value):
        self.embeddings.word_embeddings = value