From 9c67196b83a824df577742d32d38e9121d8a9285 Mon Sep 17 00:00:00 2001 From: Lysandre Date: Tue, 4 Feb 2020 11:11:37 -0500 Subject: [PATCH] Update quickstart --- docs/source/quickstart.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/quickstart.md b/docs/source/quickstart.md index 60e2cf3fd8..cdcc3b6bb0 100644 --- a/docs/source/quickstart.md +++ b/docs/source/quickstart.md @@ -299,8 +299,8 @@ model = Model2Model.from_pretrained('fine-tuned-weights') model.eval() # If you have a GPU, put everything on cuda -question_tensor = encoded_question.to('cuda') -answer_tensor = encoded_answer.to('cuda') +question_tensor = question_tensor.to('cuda') +answer_tensor = answer_tensor.to('cuda') model.to('cuda') # Predict all tokens