def qa_s2s_generate()

in longform-qa/lfqa_utils.py [0:0]

• 14 lines of code
• 1 McCabe index (conditional complexity)

def qa_s2s_generate(
    question_doc,
    qa_s2s_model,
    qa_s2s_tokenizer,
    num_answers=1,
    num_beams=None,
    min_len=64,
    max_len=256,
    do_sample=False,
    temp=1.0,
    top_p=None,
    top_k=None,
    max_input_length=512,
    device="cuda:0",