ImportError: невозможно импортировать имя hf_bucket_url в HuggingFace Transformers

Question

Итак, я установил последнюю версию трансформаторов на Google Colab

!pip install transformers 

При попытке вызвать файл преобразования с помощью

!python /usr/local/lib/python3.6/dist-packages/transformers/convert_pytorch_checkpoint_to_tf2.py .py --help  

Или пытаюсь использовать

from transformers.file_utils import hf_bucket_url.                                 // works 
from transformers.convert_pytorch_checkpoint_to_tf2 import *.                      // fails

convert_pytorch_checkpoint_to_tf("gpt2", pytorch_file, config_file, tf_file).      

Я получаю эту ошибку

 ImportError                               Traceback (most recent call last)

<ipython-input-3-dadaf83ecea0> in <module>()
      1 from transformers.file_utils import hf_bucket_url
----> 2 from transformers.convert_pytorch_checkpoint_to_tf2 import *
      3 
      4 convert_pytorch_checkpoint_to_tf("gpt2", pytorch_file, config_file, tf_file)
/usr/local/lib/python3.6/dist-packages/transformers/convert_pytorch_checkpoint_to_tf2.py in <module>()
     20 import os
     21 
---> 22 from transformers import (
     23     ALBERT_PRETRAINED_CONFIG_ARCHIVE_MAP,
     24     BERT_PRETRAINED_CONFIG_ARCHIVE_MAP,

ImportError: cannot import name 'hf_bucket_url'

В чем дело?

Answer 1

Оказывается, это ошибка. Этот PR решает проблему путем импорта функцииhf_bucket_url должным образом.