From d9fcd70ca68f18a01df1cdf828cef1a02f6de2d7 Mon Sep 17 00:00:00 2001 From: Saiwing Yeung Date: Thu, 25 Nov 2021 10:07:11 -0800 Subject: [PATCH] Update interact.py for new version of transformers https://github.com/huggingface/transfer-learning-conv-ai/issues/98 --- interact.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/interact.py b/interact.py index d368204..5e3cb06 100644 --- a/interact.py +++ b/interact.py @@ -66,7 +66,7 @@ def sample_sequence(personality, history, tokenizer, model, args, current_output input_ids = torch.tensor(instance["input_ids"], device=args.device).unsqueeze(0) token_type_ids = torch.tensor(instance["token_type_ids"], device=args.device).unsqueeze(0) - logits = model(input_ids, token_type_ids=token_type_ids) + logits = model(input_ids, token_type_ids=token_type_ids).logits if isinstance(logits, tuple): # for gpt2 and maybe others logits = logits[0] logits = logits[0, -1, :] / args.temperature