Skip to content

GitLab

Commit 9c67196b authored by Lysandre's avatar Lysandre
Browse files

Update quickstart

parent 90ab15cb
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
docs/source/quickstart.md
View file @ 9c67196b
...@@ -299,8 +299,8 @@ model = Model2Model.from_pretrained('fine-tuned-weights') ...@@ -299,8 +299,8 @@ model = Model2Model.from_pretrained('fine-tuned-weights')
model.eval() model.eval()
# If you have a GPU, put everything on cuda # If you have a GPU, put everything on cuda
question_tensor = encoded_question.to('cuda') question_tensor = question_tensor.to('cuda')
answer_tensor = encoded_answer.to('cuda') answer_tensor = answer_tensor.to('cuda')
model.to('cuda') model.to('cuda')
# Predict all tokens # Predict all tokens
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment