diff --git a/api/onnx_web/diffusers/pipelines/controlnet.py b/api/onnx_web/diffusers/pipelines/controlnet.py index e0515dc5..5dc8e01c 100644 --- a/api/onnx_web/diffusers/pipelines/controlnet.py +++ b/api/onnx_web/diffusers/pipelines/controlnet.py @@ -41,6 +41,7 @@ class OnnxStableDiffusionControlNetPipeline(DiffusionPipeline): unet: OnnxRuntimeModel, controlnet: OnnxRuntimeModel, scheduler: Union[DDIMScheduler, PNDMScheduler, LMSDiscreteScheduler], + requires_safety_checker: bool = False, ): super().__init__()