diff --git a/ppdiffusers/examples/community/webui_stable_diffusion.py b/ppdiffusers/examples/community/webui_stable_diffusion.py index 5a2522b7d32173..7ff8b766dbfee8 100644 --- a/ppdiffusers/examples/community/webui_stable_diffusion.py +++ b/ppdiffusers/examples/community/webui_stable_diffusion.py @@ -378,11 +378,11 @@ def __init__( text_encoder: CLIPTextModel, tokenizer: CLIPTokenizer, unet: UNet2DConditionModel, - controlnet: Union[ControlNetModel, List[ControlNetModel], Tuple[ - ControlNetModel], MultiControlNetModel], scheduler: KarrasDiffusionSchedulers, safety_checker: StableDiffusionSafetyChecker, feature_extractor: CLIPFeatureExtractor, + controlnet: Union[ControlNetModel, List[ControlNetModel], Tuple[ + ControlNetModel], MultiControlNetModel] = None, requires_safety_checker: bool=True, ): super().__init__()