Update README.md
Browse files
README.md
CHANGED
@@ -36,7 +36,8 @@ Inference:
|
|
36 |
tokenizer = AutoTokenizer.from_pretrained("notdiamond/notdiamond-0001")
|
37 |
model = AutoModelForSequenceClassification.from_pretrained("notdiamond/notdiamond-0001")
|
38 |
|
39 |
-
inputs = tokenizer(formatted_prompt,
|
|
|
40 |
logits = model(**inputs).logits
|
41 |
|
42 |
model_id = logits.argmax().item()
|
|
|
36 |
tokenizer = AutoTokenizer.from_pretrained("notdiamond/notdiamond-0001")
|
37 |
model = AutoModelForSequenceClassification.from_pretrained("notdiamond/notdiamond-0001")
|
38 |
|
39 |
+
inputs = tokenizer(formatted_prompt,
|
40 |
+
truncation=True, max_length=4096, return_tensors="pt")
|
41 |
logits = model(**inputs).logits
|
42 |
|
43 |
model_id = logits.argmax().item()
|