forgot sd3 inpaint

huggingface · Nov 19, 2024 · 417db60 · 417db60
1 parent 2183fb7
commit 417db60
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/optimum/onnxruntime/modeling_diffusion.py b/optimum/onnxruntime/modeling_diffusion.py
@@ -1075,6 +1075,7 @@ def _get_ort_class(pipeline_class_name: str, throw_error_if_not_exist: bool = Tr
 ORT_INPAINT_PIPELINES_MAPPING = OrderedDict(
     [
         ("stable-diffusion", ORTStableDiffusionInpaintPipeline),
+        ("stable-diffusion-3", ORTStableDiffusion3InpaintPipeline),
         ("stable-diffusion-xl", ORTStableDiffusionXLInpaintPipeline),
     ]
 )