src/transformers/models/switch_transformers/modeling_switch_transformers.py/SwitchTransformersStack/get_input_embeddings
class SwitchTransformersStack:    def get_input_embeddings(self):
        return self.embed_tokens