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