Update convert_pytorch_checkpoint_to_tf2.py (#5531)

fixed ImportError: cannot import name 'hf_bucket_url'

Update convert_pytorch_checkpoint_to_tf2.py (#5531)
fixed ImportError: cannot import name 'hf_bucket_url'
1bc13697 · Mohamed Taher Alrefaie · GitHub · b2309cc6 · 1bc13697
Unverified Commit 1bc13697 authored Jul 06, 2020 by Mohamed Taher Alrefaie Committed by GitHub Jul 06, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/convert_pytorch_checkpoint_to_tf2.py src/transformers/convert_pytorch_checkpoint_to_tf2.py +1 -1

No files found.
--- a/src/transformers/convert_pytorch_checkpoint_to_tf2.py
+++ b/src/transformers/convert_pytorch_checkpoint_to_tf2.py
@@ -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