Skip to content

GitLab

Commit 81c7e3ec authored by thomwolf's avatar thomwolf
Browse files

fix typo in readme

parent e8fe6b71
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
README.md
View file @ 81c7e3ec
...@@ -350,7 +350,7 @@ with torch.no_grad(): ...@@ -350,7 +350,7 @@ with torch.no_grad():
# get the predicted last token # get the predicted last token
predicted_index = torch.argmax(predictions_2[0, -1, :]).item() predicted_index = torch.argmax(predictions_2[0, -1, :]).item()
predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])[0] predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])[0]
assert predicted_token == '.</w>' assert predicted_token == 'who'
``` ```
## Doc ## Doc
......
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