Update quickstart

2020-02-04 11:11:37 -05:00
parent 90ab15cb7a
commit 9c67196b83
1 changed files with 2 additions and 2 deletions
--- 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