fix import

huggingface · Oct 15, 2024 · 222cbd2 · 222cbd2
1 parent e06d47a
commit 222cbd2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/onnxruntime/modeling_diffusion.py b/optimum/onnxruntime/modeling_diffusion.py
@@ -51,6 +51,7 @@
 from transformers.modeling_outputs import ModelOutput
 
 import onnxruntime as ort
+from optimum.utils import check_if_diffusers_greater
 
 from ..exporters.onnx import main_export
 from ..onnx.utils import _get_model_external_data_paths
@@ -65,7 +66,6 @@
 from .modeling_ort import ONNX_MODEL_END_DOCSTRING, ORTModel
 from .utils import (
     ONNX_WEIGHTS_NAME,
-    check_if_diffusers_greater,
     get_provider_for_device,
     np_to_pt_generators,
     parse_device,