fix(api): add missing param to controlnet pipeline

ssube · Dec 19, 2023 · 1163acc · 1163acc
1 parent a0ee620
commit 1163acc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/api/onnx_web/diffusers/pipelines/controlnet.py b/api/onnx_web/diffusers/pipelines/controlnet.py
@@ -41,6 +41,7 @@ def __init__(
         unet: OnnxRuntimeModel,
         controlnet: OnnxRuntimeModel,
         scheduler: Union[DDIMScheduler, PNDMScheduler, LMSDiscreteScheduler],
+        requires_safety_checker: bool = False,
     ):
         super().__init__()