From b3d5cccd5389a45509f3f728101a91be93c2916b Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Tue, 18 Jul 2023 15:40:21 +0200 Subject: [PATCH] add comment --- optimum/onnxruntime/modeling_diffusion.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/optimum/onnxruntime/modeling_diffusion.py b/optimum/onnxruntime/modeling_diffusion.py index 8a7b686f53f..008987cf1cb 100644 --- a/optimum/onnxruntime/modeling_diffusion.py +++ b/optimum/onnxruntime/modeling_diffusion.py @@ -547,6 +547,8 @@ def __call__(self, *args, **kwargs): class ORTStableDiffusionXLPipelineBase(ORTStableDiffusionPipelineBase): + # img2img pipeline in allows optional components + # https://github.com/huggingface/diffusers/blob/v0.18.2/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py#L111 auto_model_class = StableDiffusionXLImg2ImgPipeline def __init__(