Views
No views yet
@article{chen2021can,
title={Can NLI Models Verify QA Systems' Predictions?},
author={Chen, Jifan and Choi, Eunsol and Durrett, Greg},
journal={EMNLP Findings},
year={2021}
}{question} {seperator} {answer} where the seperator is </s>.1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3tokenizer = AutoTokenizer.from_pretrained('domenicrosati/question_converter-3b')
4model = AutoModelForSeq2SeqLM.from_pretrained('domenicrosati/question_converter-3b')
5
6question = "Where in the world is Carmen Sandiego?"
7answer = "She is in Abruzzo"
8
9prompt = f'{question} </s> {answer}'
10input_ids = tokenizer(prompt, return_tensors='pt').input_ids
11output_ids = model.generate(input_ids)
12responses = tokenizer.batch_decode(output_ids, skip_special_tokens=True)['Carmen Sandiego is in Abruzzo.']