from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained(
    "your-username/affine-custom-model-v1",
    torch_dtype="auto",
    device_map="auto"
)
tokenizer = AutoTokenizer.from_pretrained("your-username/affine-custom-model-v1")

messages = [{"role": "user", "content": "Solve this SAT problem..."}]
inputs = tokenizer.apply_chat_template(messages, return_tensors="pt")
outputs = model.generate(inputs, max_new_tokens=512)
print(tokenizer.decode(outputs[0]))

Performance

This model achieves competitive performance across all Affine evaluation environments.

License

Apache 2.0

mateo123123/ys-v2

your-username/affine-custom-model-v1

Model Details

Environments

Usage

Performance

License

mateo123123/
ys-v2