Skip to content

Commit f738274

Browse files
authored
Merge pull request #1 from lxp521125/lxp521125-patch-1
Update webui_stable_diffusion.py通过WebUIStableDiffusionPipeline.from_pretrained_original_ckpt加载模型因为contronet报错修复
2 parents ebb3f5b + 80c4d88 commit f738274

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ppdiffusers/examples/community/webui_stable_diffusion.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -378,11 +378,11 @@ def __init__(
378378
text_encoder: CLIPTextModel,
379379
tokenizer: CLIPTokenizer,
380380
unet: UNet2DConditionModel,
381-
controlnet: Union[ControlNetModel, List[ControlNetModel], Tuple[
382-
ControlNetModel], MultiControlNetModel],
383381
scheduler: KarrasDiffusionSchedulers,
384382
safety_checker: StableDiffusionSafetyChecker,
385383
feature_extractor: CLIPFeatureExtractor,
384+
controlnet: Union[ControlNetModel, List[ControlNetModel], Tuple[
385+
ControlNetModel], MultiControlNetModel] = None,
386386
requires_safety_checker: bool=True, ):
387387
super().__init__()
388388

0 commit comments

Comments
 (0)