From f2d0f8410c587ba502d114c5faca59ea8d23fcd7 Mon Sep 17 00:00:00 2001
From: Ella Charlaix <ella@huggingface.co>
Date: Fri, 15 Sep 2023 17:14:46 +0200
Subject: [PATCH] format

---
 optimum/onnxruntime/modeling_decoder.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/optimum/onnxruntime/modeling_decoder.py b/optimum/onnxruntime/modeling_decoder.py
index 64c7b9bd204..8e484b5d8ea 100644
--- a/optimum/onnxruntime/modeling_decoder.py
+++ b/optimum/onnxruntime/modeling_decoder.py
@@ -23,10 +23,12 @@
 import torch
 from huggingface_hub import hf_hub_download
 from huggingface_hub.utils import EntryNotFoundError
+from onnx.tools import update_model_dims
 from transformers import AutoModelForCausalLM, GenerationConfig
 from transformers.file_utils import add_end_docstrings, add_start_docstrings_to_model_forward
 from transformers.modeling_outputs import CausalLMOutputWithPast
 
+import onnx
 import onnxruntime
 
 from ..exporters import TasksManager
@@ -50,8 +52,6 @@
     validate_provider_availability,
 )
 
-import onnx
-from onnx.tools import update_model_dims
 
 if TYPE_CHECKING:
     from transformers import PretrainedConfig