src/transformers/models/decision_transformer/modeling_decision_transformer.py/DecisionTransformerGPT2Model/set_input_embeddings
class DecisionTransformerGPT2Model:    def set_input_embeddings(self, new_embeddings):
        self.wte = new_embeddings