Update convert_pytorch_checkpoint_to_tf2.py (#5531)

fixed ImportError: cannot import name 'hf_bucket_url'
This commit is contained in:
Mohamed Taher Alrefaie 2020-07-06 15:55:10 +02:00 committed by GitHub
parent b2309cc6bf
commit 1bc13697b1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -71,11 +71,11 @@ from transformers import (
XLMRobertaConfig,
XLNetConfig,
cached_path,
hf_bucket_url,
is_torch_available,
load_pytorch_checkpoint_in_tf2_model,
)
from transformers.file_utils import hf_bucket_url
if is_torch_available():
import torch