From d181ad22e482b8e26ca172981f48591957f7acdd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?F=C3=A9lix=20Marty?= <9808326+fxmarty@users.noreply.github.com> Date: Thu, 21 Sep 2023 14:56:27 +0200 Subject: [PATCH] nit^2 --- optimum/exporters/onnx/model_configs.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py index 51d5823774f..d339f565207 100644 --- a/optimum/exporters/onnx/model_configs.py +++ b/optimum/exporters/onnx/model_configs.py @@ -56,13 +56,12 @@ VisionOnnxConfig, ) from .model_patcher import SAMModelPatcher, SpeechT5ModelPatcher, WavLMModelPatcher - +from ...utils import DummyInputGenerator if TYPE_CHECKING: from transformers import PretrainedConfig from transformers.modeling_utils import PreTrainedModel - from ...utils import DummyInputGenerator from .model_patcher import ModelPatcher if is_tf_available():