From 81c7e3ec9f26e774902276769d32140bc699c631 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Mon, 11 Feb 2019 13:37:12 +0100 Subject: [PATCH] fix typo in readme --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 8265c5d246..4bce27db47 100644 --- a/README.md +++ b/README.md @@ -350,7 +350,7 @@ with torch.no_grad(): # get the predicted last token predicted_index = torch.argmax(predictions_2[0, -1, :]).item() predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])[0] -assert predicted_token == '.' +assert predicted_token == 'who' ``` ## Doc