apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): add missing param to controlnet pipeline

This commit is contained in:
Sean Sube 2023-12-18 21:39:49 -06:00
parent a0ee620336
commit 1163accb49
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/onnx_web/diffusers/pipelines/controlnet.py
View File

@ -41,6 +41,7 @@ class OnnxStableDiffusionControlNetPipeline(DiffusionPipeline):
unet: OnnxRuntimeModel, unet: OnnxRuntimeModel,
controlnet: OnnxRuntimeModel, controlnet: OnnxRuntimeModel,
scheduler: Union[DDIMScheduler, PNDMScheduler, LMSDiscreteScheduler], scheduler: Union[DDIMScheduler, PNDMScheduler, LMSDiscreteScheduler],
requires_safety_checker: bool = False,
): ):
super().__init__() super().__init__()