diff --git a/invokeai/backend/model_management/models/stable_diffusion_onnx.py b/invokeai/backend/model_management/models/stable_diffusion_onnx.py
index c678b27646..2780ba4728 100644
--- a/invokeai/backend/model_management/models/stable_diffusion_onnx.py
+++ b/invokeai/backend/model_management/models/stable_diffusion_onnx.py
@@ -1,5 +1,7 @@
 from enum import Enum
 from typing import Literal
+
+from diffusers import OnnxRuntimeModel
 from .base import (
     ModelConfigBase,
     BaseModelType,
@@ -8,7 +10,6 @@ from .base import (
     DiffusersModel,
     SchedulerPredictionType,
     classproperty,
-    OnnxRuntimeModel,
     IAIOnnxRuntimeModel,
 )