From 1163accb49cdb45f5577b2eca31fba6cb539b37b Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 18 Dec 2023 21:39:49 -0600 Subject: [PATCH] fix(api): add missing param to controlnet pipeline --- api/onnx_web/diffusers/pipelines/controlnet.py | 1 + 1 file changed, 1 insertion(+) 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__()