Views
No views yet
jhu-clsp/ettin-encoder-400m (ModernBERT architecture)transformers library:1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4model_name = "usmanqamr/math-misunderstanding-ettin-v1"
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForSequenceClassification.from_pretrained(model_name)
7
8text = "Question: What is 1/2 + 1/3? Student Answer: 2/5"
9inputs = tokenizer(text, return_tensors="pt")
10
11with torch.no_grad():
12 logits = model(**inputs).logits
13
14predicted_class = torch.argmax(logits, dim=-1)
15print(predicted_class)