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