Views
No views yet
config.jsonmodel.safetensorstokenizer.jsontokenizer_config.jsonspecial_tokens_map.json1import torch
2from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
4model_path = "."
5tokenizer = AutoTokenizer.from_pretrained(model_path)
6model = AutoModelForSequenceClassification.from_pretrained(model_path)
7model.eval()
8
9texts = [
10 "I disagree with this argument, but the evidence is interesting.",
11 "You are awful and should leave.",
12]
13
14inputs = tokenizer(
15 texts,
16 padding=True,
17 truncation=True,
18 return_tensors="pt",
19)
20
21with torch.no_grad():
22 logits = model(**inputs).logits
23 scores = torch.sigmoid(logits)
24
25print(scores.tolist())