OddEvenDumb / README.md
56m's picture
Update README.md
a3b5e0b verified
|
Raw
History Blame Contribute Delete
569 Bytes
metadata
license: apache-2.0

OddEvenDumb

What this AI can do

  • OddEvenDumb can check whether A binary is even or odd. That's all.

How to run

from transformers import AutoModelForCausalLM, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("56m/OddEvenDumb")
model = AutoModelForCausalLM.from_pretrained( 
"56m/OddEvenDumb", trust_remote_code=True
)

inputs = tokenizer("<BOS>1111010011", return_tensors="pt")
outputs = model(**inputs)
pred_id = outputs.logits[0, -1, :].argmax().item()
print("Result:", tokenizer.decode([pred_id])) # '1'