Update README.md

cc @patil-suraj

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")
+config = RobertaConfig.from_pretrained("roberta-base", vocab_size=tokenizer.vocab_size)
 config.save_pretrained(model_dir)
 ```
 
@@ -211,7 +211,7 @@ from transformers import GPT2Config
 
 model_dir = "./norwegian-gpt2"  # ${MODEL_DIR}
 
-config = GPT2Config.from_pretrained("gpt2", resid_pdrop=0.0, embd_pdrop=0.0, attn_pdrop=0.0)
+config = GPT2Config.from_pretrained("gpt2", resid_pdrop=0.0, embd_pdrop=0.0, attn_pdrop=0.0, vocab_size=tokenizer.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")
+config = T5Config.from_pretrained("google/t5-v1_1-base", vocab_size=tokenizer.vocab_size)
 config.save_pretrained(model_dir)
 ```