How to use edbeeching/decision-transformer-gym-walker2d-medium with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("edbeeching/decision-transformer-gym-walker2d-medium") model = AutoModel.from_pretrained("edbeeching/decision-transformer-gym-walker2d-medium")