Transformers How to use TeeA/roberta-base-pokemon with Transformers:
# Load model directly
from transformers import RobertaModelForTextClassification
model = RobertaModelForTextClassification.from_pretrained("TeeA/roberta-base-pokemon", device_map="auto") 
