src/transformers/models/flava/modeling_flava.py/FlavaTextModel/set_input_embeddings
class FlavaTextModel:    def set_input_embeddings(self, value: nn.Module):
        self.embeddings.word_embeddings = value