From aa438a4265e848d40a8b7099dde7152387b07a8a Mon Sep 17 00:00:00 2001 From: Stefan Schweter Date: Tue, 2 Feb 2021 12:09:24 +0100 Subject: [PATCH] convbert: minor fixes for conversion script (#9937) --- .../convert_convbert_original_tf1_checkpoint_to_pytorch.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/transformers/models/convbert/convert_convbert_original_tf1_checkpoint_to_pytorch.py b/src/transformers/models/convbert/convert_convbert_original_tf1_checkpoint_to_pytorch.py index 7bb8a6ce51..aaeb77784e 100644 --- a/src/transformers/models/convbert/convert_convbert_original_tf1_checkpoint_to_pytorch.py +++ b/src/transformers/models/convbert/convert_convbert_original_tf1_checkpoint_to_pytorch.py @@ -16,8 +16,8 @@ import argparse -from ...utils import logging -from .modeling_convbert import ConvBertConfig, ConvBertModel, load_tf_weights_in_convbert +from transformers import ConvBertConfig, ConvBertModel, load_tf_weights_in_convbert +from transformers.utils import logging logging.set_verbosity_info() @@ -49,4 +49,4 @@ if __name__ == "__main__": "--pytorch_dump_path", default=None, type=str, required=True, help="Path to the output PyTorch model." ) args = parser.parse_args() - convert_orig_tf1_checkpoint_to_pytorch(args.tf_checkpoint_path, args.conv_bert_config_file, args.pytorch_dump_path) + convert_orig_tf1_checkpoint_to_pytorch(args.tf_checkpoint_path, args.convbert_config_file, args.pytorch_dump_path)