Я попытался использовать Convert_graph_to_onnx.py из Hugginsface, но получил следующую ошибку: «Ошибка при преобразовани ⇐ Python

[Anonymous]

Я попытался использовать функцию экспорта модели, используя пример на этой [странице][1]

Код: Выделить всё

python convert_graph_to_onnx.py --framework pt --model bert-base-cased bert-base-cased.onnx

Я получил ошибку, что мне нужно использовать ONNX opset версии 14, поэтому я использовал это:

Код: Выделить всё

python convert_graph_to_onnx.py --framework pt --opset 14 --model bert-base-cased bert-base-cased.onnx

И я получил эту ошибку:

====== Преобразование модели в ONNX ===== = Convert_graph_to_onnx.py:361: FutureWarning: Пакет 'transformers.convert_graph_to_onnx`
устарел и будет удален в версии 5 Transformers'

warnings.warn (версия ONNX opset установлена ​​на: 14 Конвейер загрузки (модель:
bert-base-cased, токенизатор: bert-base-cased) Использование платформы PyTorch:
2.3.1+cpu Обнаружены входные input_ids с формой: {0: 'batch', 1: 'sequence'} Найдены входные token_type_ids с формой: {0: 'batch', 1:
'sequence'} Найден входной объект внимания_маска с формой: {0: 'batch', 1:
'sequence '} Найден выходной вывод_0 с формой: {0: 'batch', 1:
'sequence'} Найден выходной вывод_1 с формой: {0: 'batch'} Проверка
входных данных в правильном порядке отсутствует в
сгенерированном списке входных данных. Порядок сгенерированных входных данных: ['input_ids',
'attention_mask', 'token_type_ids'] Ошибка при преобразовании модели:
Модуль onnx не установлен!< /p>

У меня установлен onnx==1.16.1
Может ли кто-нибудь мне помочь?

Подробнее здесь: https://stackoverflow.com/questions/786 ... -this-erro