How to use diamond0/dummy-model with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("diamond0/dummy-model") model = AutoModelForSequenceClassification.from_pretrained("diamond0/dummy-model", device_map="auto")
This is the dummy model following the huggingface NLP course.
