Update README.md

README.md

@@ -35,5 +35,5 @@ logits = model(input_ids).logits
 next_token_logits = logits[-1, :]
 next_token_probs = F.softmax(next_token_logits, dim=-1)
 for i in range(tokenizer.get_vocab_size(with_added_tokens=False)):
-    print(f"{tokenizer.id_to_token(i)}: {round(100 * next_token_probs[i].item(), 2):.2f} %")
+    print(f"{tokenizer.id_to_token(i)}: {100 * next_token_probs[i].item():.2f} %")
 ```