From 80c4d88dc4e00dd9d9c5ad39b866485ba3c851ad Mon Sep 17 00:00:00 2001
From: Lei Xiaopeng <lxp521125@users.noreply.github.com>
Date: Tue, 8 Aug 2023 22:07:25 +0800
Subject: [PATCH] Update webui_stable_diffusion.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

当使用StableDiffusionPipeline.from_pretrained_original_ckpt()
导入safetensors会因为没有controlnet报错，可以给个默认值None，底层已经处理过None。
---
 ppdiffusers/examples/community/webui_stable_diffusion.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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__()