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

Итак, я установил последнюю версию трансформеров на 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'

В чем дело?


person Mohamed Taher Alrefaie    schedule 05.07.2020    source источник
comment
Некоторая проблема появилась в версии 2.11. Откройте отчет об ошибке и в качестве временного решения вернитесь к 2.10.0   -  person cronoik    schedule 06.07.2020
comment
Я только что отправил PR .. github.com/huggingface/transformers/pull/5531   -  person Mohamed Taher Alrefaie    schedule 06.07.2020


Ответы (1)


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

person Mohamed Taher Alrefaie    schedule 07.07.2020