why wouldn't this be fine?

huggingface · Aug 1, 2023 · b52ac54 · b52ac54
1 parent 26c7549
commit b52ac54
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
+[[autodoc]] onnxruntime.ORTStableDiffusionXLImg2ImgPipeline
diff --git 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):