From 53023c515c69f04df2d7b383627839185a7ca421 Mon Sep 17 00:00:00 2001 From: Pzzzzz Date: Fri, 24 May 2024 13:31:57 +0800 Subject: [PATCH] Fix nmt weight conversion --- examples/enc_dec/convert_checkpoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/enc_dec/convert_checkpoint.py b/examples/enc_dec/convert_checkpoint.py index 5845b0045..a0e2d607f 100755 --- a/examples/enc_dec/convert_checkpoint.py +++ b/examples/enc_dec/convert_checkpoint.py @@ -331,7 +331,7 @@ def parse_nmt_config(args, model): config["decoder"]["q_scaling"] = '1' config["decoder"]["rescale_before_lm_head"] = 'false' config['decoder']['has_model_final_layernorm'] = 'false' - config['decoder']['vocab_size'] = str(len(model.src_dict)) # fairseq naming + config['decoder']['vocab_size'] = str(len(model.tgt_dict)) # fairseq naming config["structure"] = dict() config["structure"]["t5_with_bias"] = "true"