| --- |
| 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 |
| ```python |
| 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' |
| ```` |
