From d026fdc42262c88bc8f4b004325212c120aab4e4 Mon Sep 17 00:00:00 2001 From: fxmarty <9808326+fxmarty@users.noreply.github.com> Date: Wed, 7 Dec 2022 11:10:57 +0100 Subject: [PATCH] fix import (#553) --- optimum/exporters/onnx/__main__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/optimum/exporters/onnx/__main__.py b/optimum/exporters/onnx/__main__.py index d97116143c5..96c35161c76 100644 --- a/optimum/exporters/onnx/__main__.py +++ b/optimum/exporters/onnx/__main__.py @@ -20,7 +20,7 @@ from transformers import AutoTokenizer from ...utils import logging -from ...utils.save_utils import maybe_save_tokenizer_or_processor_or_feature_extractor +from ...utils.save_utils import maybe_save_preprocessors from ..tasks import TasksManager from .base import OnnxConfigWithPast from .convert import ( @@ -144,7 +144,7 @@ def main(): # Saving the model config as this is needed sometimes. model.config.save_pretrained(args.output.parent) - maybe_save_tokenizer_or_processor_or_feature_extractor(args.model, args.output.parent) + maybe_save_preprocessors(args.model, args.output.parent) if args.atol is None: args.atol = onnx_config.ATOL_FOR_VALIDATION