Create README.md
Browse files
README.md
CHANGED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
model_name = 'logicreasoning/LogiT5'
|
2 |
+
tokenize = AutoTokenizer.from_pretrained(model_name)
|
3 |
+
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
4 |
+
device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
|
5 |
+
input_text = '' #your input text here
|
6 |
+
input = tokenize(input, return_tensors='pt', padding=True).to(device)
|
7 |
+
model = model.to(device)
|
8 |
+
output = model.generate(*input, max_length=1024)
|
9 |
+
prediction = tokenize.decode(output[0],skip_special_tokens=True)
|