saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix GPT conversion script (#9676)

Browse source
This commit is contained in:
Sylvain Gugger 2021-01-19 09:55:37 -05:00 committed by Lysandre
parent c5f6719040
commit 0ed8bccc9c
1 changed files with 3 additions and 3 deletions
Show stats 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 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()