Fix imports in conversion scripts (#9674)

This commit is contained in:
Sylvain Gugger
2021-01-19 09:40:15 -05:00
committed by GitHub
parent 2390c16fd2
commit 053efc5d2d
28 changed files with 75 additions and 63 deletions

View File

@@ -28,8 +28,8 @@ import re
import tensorflow as tf
import torch
from ...utils import logging
from . import BertConfig, BertModel
from transformers import BertConfig, BertModel
from transformers.utils import logging
logging.set_verbosity_info()

View File

@@ -19,8 +19,8 @@ import argparse
import torch
from ...utils import logging
from . import BertConfig, BertForPreTraining, load_tf_weights_in_bert
from transformers import BertConfig, BertForPreTraining, load_tf_weights_in_bert
from transformers.utils import logging
logging.set_verbosity_info()

View File

@@ -22,7 +22,7 @@ import numpy as np
import tensorflow as tf
import torch
from . import BertModel
from transformers import BertModel
def convert_pytorch_checkpoint_to_tf(model: BertModel, ckpt_dir: str, model_name: str):