fix(api): add missing param to controlnet pipeline
This commit is contained in:
parent
a0ee620336
commit
1163accb49
|
@ -41,6 +41,7 @@ class OnnxStableDiffusionControlNetPipeline(DiffusionPipeline):
|
|||
unet: OnnxRuntimeModel,
|
||||
controlnet: OnnxRuntimeModel,
|
||||
scheduler: Union[DDIMScheduler, PNDMScheduler, LMSDiscreteScheduler],
|
||||
requires_safety_checker: bool = False,
|
||||
):
|
||||
super().__init__()
|
||||
|
||||
|
|
Loading…
Reference in New Issue