From b52ac5403719ba40c7d9396e5ab3156e5b5380ab Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?F=C3=A9lix=20Marty?=
 <9808326+fxmarty@users.noreply.github.com>
Date: Tue, 1 Aug 2023 14:37:50 +0200
Subject: [PATCH] why wouldn't this be fine?

---
 docs/source/onnxruntime/package_reference/modeling_ort.mdx | 3 ++-
 optimum/onnxruntime/modeling_diffusion.py                  | 5 ++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/docs/source/onnxruntime/package_reference/modeling_ort.mdx b/docs/source/onnxruntime/package_reference/modeling_ort.mdx
index ebbfa1736e..3d3daa8865 100644
--- a/docs/source/onnxruntime/package_reference/modeling_ort.mdx
+++ b/docs/source/onnxruntime/package_reference/modeling_ort.mdx
@@ -113,6 +113,7 @@ The following ORT classes are available for the following custom tasks.
 #### ORTStableDiffusionPipeline
 
 [[autodoc]] onnxruntime.ORTStableDiffusionPipeline
+    - __call__
 
 #### ORTStableDiffusionImg2ImgPipeline
 
@@ -129,4 +130,4 @@ The following ORT classes are available for the following custom tasks.
 
 #### ORTStableDiffusionXLImg2ImgPipeline
 
-[[autodoc]] onnxruntime.ORTStableDiffusionXLImg2ImgPipeline
\ No newline at end of file
+[[autodoc]] onnxruntime.ORTStableDiffusionXLImg2ImgPipeline
diff --git a/optimum/onnxruntime/modeling_diffusion.py b/optimum/onnxruntime/modeling_diffusion.py
index 8a7b686f53..1a84aa3d22 100644
--- a/optimum/onnxruntime/modeling_diffusion.py
+++ b/optimum/onnxruntime/modeling_diffusion.py
@@ -531,9 +531,8 @@ def forward(self, sample: np.ndarray):
         return outputs
 
 
-class ORTStableDiffusionPipeline(ORTStableDiffusionPipelineBase, StableDiffusionPipelineMixin):
-    def __call__(self, *args, **kwargs):
-        return StableDiffusionPipelineMixin.__call__(self, *args, **kwargs)
+class ORTStableDiffusionPipeline(StableDiffusionPipelineMixin, ORTStableDiffusionPipelineBase):
+    pass
 
 
 class ORTStableDiffusionImg2ImgPipeline(ORTStableDiffusionPipelineBase, StableDiffusionImg2ImgPipelineMixin):