Transformers How to use KotshinZ/gpt2-RMT-2 with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import RecurrentMemoryTransformer
model = RecurrentMemoryTransformer.from_pretrained("KotshinZ/gpt2-RMT-2", device_map="auto") 
