examples: use correct way to get vocab size in flax lm readme (#12947)

3d4b3bc3 · Stefan Schweter · GitHub · 23d6761f · 3d4b3bc3
Unverified Commit 3d4b3bc3 authored Jul 30, 2021 by Stefan Schweter Committed by GitHub Jul 30, 2021
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

examples/flax/language-modeling/README.md examples/flax/language-modeling/README.md +2 -2

No files found.
--- a/examples/flax/language-modeling/README.md
+++ b/examples/flax/language-modeling/README.md
@@ -114,7 +114,7 @@ from transformers import RobertaConfig

 model_dir = "./norwegian-roberta-base"  # ${MODEL_DIR}

-config = RobertaConfig.from_pretrained("roberta-base", vocab_size=tokenizer.vocab_size)
+config = RobertaConfig.from_pretrained("roberta-base", vocab_size=tokenizer.get_vocab_size())
 config.save_pretrained(model_dir)
 ```

@@ -349,7 +349,7 @@ from transformers import T5Config

 model_dir = "./norwegian-t5-base"  # ${MODEL_DIR}

-config = T5Config.from_pretrained("google/t5-v1_1-base", vocab_size=tokenizer.vocab_size)
+config = T5Config.from_pretrained("google/t5-v1_1-base", vocab_size=tokenizer.get_vocab_size())
 config.save_pretrained(model_dir)
 ```