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