Skip to content

GitLab

Unverified Commit d302d88b authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix GPT conversion script (#9676)

parent 053efc5d
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
src/transformers/models/openai/convert_openai_original_tf_checkpoint_to_pytorch.py
View file @ d302d88b
...@@ -19,9 +19,9 @@ import argparse ...@@ -19,9 +19,9 @@ import argparse
import torch import torch
from ...file_utils import CONFIG_NAME, WEIGHTS_NAME from transformers import OpenAIGPTConfig, OpenAIGPTModel, load_tf_weights_in_openai_gpt
from ...utils import logging from transformers.file_utils import CONFIG_NAME, WEIGHTS_NAME
from . import OpenAIGPTConfig, OpenAIGPTModel, load_tf_weights_in_openai_gpt from transformers.utils import logging
logging.set_verbosity_info() logging.set_verbosity_info()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment