From 0ed8bccc9c62fbdfe2292335f4942d9c1502f9c3 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Tue, 19 Jan 2021 09:55:37 -0500 Subject: [PATCH] Fix GPT conversion script (#9676) --- .../convert_openai_original_tf_checkpoint_to_pytorch.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/transformers/models/openai/convert_openai_original_tf_checkpoint_to_pytorch.py b/src/transformers/models/openai/convert_openai_original_tf_checkpoint_to_pytorch.py index abe06f8eb..bb8aaa228 100755 --- a/src/transformers/models/openai/convert_openai_original_tf_checkpoint_to_pytorch.py +++ b/src/transformers/models/openai/convert_openai_original_tf_checkpoint_to_pytorch.py @@ -19,9 +19,9 @@ import argparse import torch -from ...file_utils import CONFIG_NAME, WEIGHTS_NAME -from ...utils import logging -from . import OpenAIGPTConfig, OpenAIGPTModel, load_tf_weights_in_openai_gpt +from transformers import OpenAIGPTConfig, OpenAIGPTModel, load_tf_weights_in_openai_gpt +from transformers.file_utils import CONFIG_NAME, WEIGHTS_NAME +from transformers.utils import logging logging.set_verbosity_info()