Skip to content

GitLab

"next_docs/en/git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "0807e971fefe435d87c071e00f7767e836196a9e"
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')
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
......
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