From 417db60678e1b61596d060291fbbc6358d7a6b63 Mon Sep 17 00:00:00 2001
From: IlyasMoutawwakil <moutawwakil.ilyas.tsi@gmail.com>
Date: Tue, 19 Nov 2024 11:52:02 +0100
Subject: [PATCH] forgot sd3 inpaint

---
 optimum/onnxruntime/modeling_diffusion.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/optimum/onnxruntime/modeling_diffusion.py b/optimum/onnxruntime/modeling_diffusion.py
index a7faecfcfa..79d302be44 100644
--- 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),
     ]
 )