From f1f23ad1710953e75b53a85953b018b8caceb427 Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Wed, 2 Oct 2019 19:03:32 -0400 Subject: [PATCH] fix buf in convert_pt_chkpt_to_tf2 --- transformers/convert_pytorch_checkpoint_to_tf2.py | 1 + 1 file changed, 1 insertion(+) diff --git a/transformers/convert_pytorch_checkpoint_to_tf2.py b/transformers/convert_pytorch_checkpoint_to_tf2.py index c5f7650b50..d8a48e9dcd 100644 --- a/transformers/convert_pytorch_checkpoint_to_tf2.py +++ b/transformers/convert_pytorch_checkpoint_to_tf2.py @@ -228,6 +228,7 @@ if __name__ == "__main__": convert_all_pt_checkpoints_to_tf(args.model_type.lower() if args.model_type is not None else None, args.tf_dump_path, model_shortcut_names_or_path=[args.pytorch_checkpoint_path] if args.pytorch_checkpoint_path is not None else None, + config_shortcut_names_or_path=[args.config_file] if args.config_file is not None else None, compare_with_pt_model=args.compare_with_pt_model, use_cached_models=args.use_cached_models, only_convert_finetuned_models=args.only_convert_finetuned_models)