format

huggingface · Sep 19, 2023 · d86bce6 · d86bce6
1 parent c2ec382
commit d86bce6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/optimum/exporters/onnx/model_patcher.py b/optimum/exporters/onnx/model_patcher.py
@@ -16,10 +16,11 @@
 import functools
 import inspect
 from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Union
-from ...utils.modeling_utils import _prepare_attn_mask, _prepare_decoder_attention_mask
 
 from transformers.utils import is_torch_available
 
+from ...utils.modeling_utils import _prepare_attn_mask, _prepare_decoder_attention_mask
+
 
 if is_torch_available():
     import torch