SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Stefan Schweter
2021-07-30 18:27:53 +02:00
committed by GitHub
parent 23d6761f30
commit 3d4b3bc3fd
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/flax/language-modeling/README.md
View File

@@ -114,7 +114,7 @@ from transformers import RobertaConfig
model_dir = "./norwegian-roberta-base" # ${MODEL_DIR} 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) config.save_pretrained(model_dir)
``` ```
@@ -349,7 +349,7 @@ from transformers import T5Config
model_dir = "./norwegian-t5-base" # ${MODEL_DIR} 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) config.save_pretrained(model_dir)
``` ```