fix style

huggingface · Dec 19, 2024 · f3962ca · f3962ca
1 parent be3c79c
commit f3962ca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/optimum/exporters/onnx/base.py b/optimum/exporters/onnx/base.py
@@ -311,8 +311,6 @@ def fix_dynamic_axes(
 
         # We branch here to avoid doing an unnecessary forward pass.
         if to_fix:
-            import onnx
-
             if input_shapes is None:
                 input_shapes = {}
             dummy_inputs = self.generate_dummy_inputs(framework="np", **input_shapes)

diff --git a/optimum/exporters/onnx/config.py b/optimum/exporters/onnx/config.py
@@ -19,6 +19,7 @@
 from typing import TYPE_CHECKING, Any, Dict, Iterable, List, Optional, Tuple, Union
 
 from transformers.utils import is_tf_available
+
 from ...utils import (
     DummyAudioInputGenerator,
     DummyBboxInputGenerator,
@@ -128,6 +129,7 @@ def post_process_exported_models(
         # Attempt to merge only if the decoder-only was exported separately without/with past
         if self.use_past is True and len(models_and_onnx_configs) == 2:
             from ...onnx import merge_decoders
+
             decoder_path = Path(path, onnx_files_subpaths[0])
             decoder_with_past_path = Path(path, onnx_files_subpaths[1])
             decoder_merged_path = Path(path, ONNX_DECODER_MERGED_NAME + ".onnx")