SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix GPT conversion script (#9676)

Browse Source
This commit is contained in:
Sylvain Gugger
2021-01-19 09:55:37 -05:00
committed by GitHub
parent 053efc5d2d
commit d302d88b47
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/models/openai/convert_openai_original_tf_checkpoint_to_pytorch.py
View File

@@ -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()