Transformers How to use aws-ai/dse-distilbert-base with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, PSCDistilBERT
tokenizer = AutoTokenizer.from_pretrained("aws-ai/dse-distilbert-base")
model = PSCDistilBERT.from_pretrained("aws-ai/dse-distilbert-base", device_map="auto") 
