Merge pull request PaddlePaddle#1 from lxp521125/lxp521125-patch-1

Update webui_stable_diffusion.py通过WebUIStableDiffusionPipeline.from_pretrained_original_ckpt加载模型因为contronet报错修复
WAYKEN-TSE · Aug 8, 2023 · f738274 · f738274
2 parents ebb3f5b + 80c4d88
commit f738274
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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__()