Skip to content
This repository was archived by the owner on Jan 15, 2024. It is now read-only.

Commit f5a9fc1

Browse files
DOUDOU0314UbuntuUbuntuUbuntuszha
authored
[BUGFIX] Fix bug in issue #1570 (#1572)
* add uncased_bert_wwm_large version * Update generate_commands.py * Update run_squad2_uncased_bert_wwm_large.sh * add wwm version * add wwm version * fixed a bug and add a NaturalQuestion index in README.md * update wikiextractor version * bug fix * modify * fix fairseq version * Update convert_fairseq_roberta.py * Update convert_fairseq_roberta.py * Update convert_fairseq_xlmr.py Co-authored-by: Ubuntu <[email protected]> Co-authored-by: Ubuntu <[email protected]> Co-authored-by: Ubuntu <[email protected]> Co-authored-by: Sheng Zha <[email protected]> Co-authored-by: Ubuntu <[email protected]>
1 parent 307d079 commit f5a9fc1

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

scripts/conversion_toolkits/convert_fairseq_xlmr.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def convert_fairseq_model(args):
8787
checkpoint_file='model.pt')
8888
vocab_size = convert_vocab(args, fairseq_xlmr)
8989

90-
gluon_cfg = convert_config(fairseq_xlmr.cfg.model, vocab_size,
90+
gluon_cfg = convert_config(fairseq_xlmr.args, vocab_size,
9191
XLMRModel.get_cfg().clone())
9292
with open(os.path.join(args.save_dir, 'model.yml'), 'w') as of:
9393
of.write(gluon_cfg.dump())

scripts/conversion_toolkits/convert_xlmr.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
python3 -m pip install git+https://github.com/pytorch/fairseq.git@master --upgrade --user
1+
python3 -m pip install fairseq==0.10.1 --upgrade --user
22
for model in base large
33
do
44
mkdir xlmr_${model}

0 commit comments

Comments
 (0)