Views
No views yet
1from transformers import AutoTokenizer, AutoModelWithLMHead
2tokenizer = AutoTokenizer.from_pretrained("dbernsohn/algebra_linear_1d_composed")
3model = AutoModelWithLMHead.from_pretrained("dbernsohn/algebra_linear_1d_composed")1query = "Suppose -d = 5 - 16. Let b = -579 + 584. Solve -b*c + 36 = d for c."
2input_text = f"{query} </s>"
3features = tokenizer([input_text], return_tensors='pt')
4model.to('cuda')
5output = model.generate(input_ids=features['input_ids'].cuda(),
6 attention_mask=features['attention_mask'].cuda())
7
8tokenizer.decode(output[0])
9# <pad> 5</s>Created by Dor Bernsohn