Smaller
Collection
<50K • 2 items • Updated
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'