File size: 604 Bytes
a01d066 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | from transformers import AutoModel, AutoTokenizer
REPO_ID = "YOUR_ORG/LaGrange-1.0-Flash"
REVISION = "main"
tokenizer = AutoTokenizer.from_pretrained(
REPO_ID,
revision=REVISION,
trust_remote_code=True,
)
model = AutoModel.from_pretrained(
REPO_ID,
revision=REVISION,
trust_remote_code=True,
)
batch = tokenizer("machine learning mathematics", return_tensors="pt")
bridge_output = model(**batch)
print(bridge_output.pooler_output.shape)
result = model.research({
"domain": "ml_math",
"action": "softmax_jacobian",
"logits": [0.2, -0.1, 1.3, 2.0],
})
print(result)
|