Update pipeline_fill_sd_xl.py
Browse files- pipeline_fill_sd_xl.py +1 -1
pipeline_fill_sd_xl.py
CHANGED
@@ -86,7 +86,7 @@ class StableDiffusionXLFillPipeline(DiffusionPipeline, StableDiffusionMixin):
|
|
86 |
tokenizer_2: CLIPTokenizer,
|
87 |
unet: UNet2DConditionModel,
|
88 |
controlnet: ControlNetModel_Union,
|
89 |
-
scheduler:
|
90 |
force_zeros_for_empty_prompt: bool = True,
|
91 |
):
|
92 |
super().__init__()
|
|
|
86 |
tokenizer_2: CLIPTokenizer,
|
87 |
unet: UNet2DConditionModel,
|
88 |
controlnet: ControlNetModel_Union,
|
89 |
+
scheduler: KarrasDiffusionSchedulers,
|
90 |
force_zeros_for_empty_prompt: bool = True,
|
91 |
):
|
92 |
super().__init__()
|