Skip to content

Commit

Permalink
add comment
Browse files Browse the repository at this point in the history
  • Loading branch information
echarlaix committed Jul 18, 2023
1 parent f26e5b1 commit b3d5ccc
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions optimum/onnxruntime/modeling_diffusion.py
Original file line number Diff line number Diff line change
Expand Up @@ -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__(
Expand Down

0 comments on commit b3d5ccc

Please sign in to comment.