make requires_safety_checker a kwarg instead of a positional argument as it's more future-proof
Browse files- pipeline.py +8 -1
pipeline.py
CHANGED
|
@@ -276,7 +276,14 @@ class LLMGroundedDiffusionPipeline(StableDiffusionPipeline):
|
|
| 276 |
requires_safety_checker: bool = True,
|
| 277 |
):
|
| 278 |
super().__init__(
|
| 279 |
-
vae,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 280 |
)
|
| 281 |
|
| 282 |
self.register_attn_hooks(unet)
|
|
|
|
| 276 |
requires_safety_checker: bool = True,
|
| 277 |
):
|
| 278 |
super().__init__(
|
| 279 |
+
vae,
|
| 280 |
+
text_encoder,
|
| 281 |
+
tokenizer,
|
| 282 |
+
unet,
|
| 283 |
+
scheduler,
|
| 284 |
+
safety_checker,
|
| 285 |
+
feature_extractor,
|
| 286 |
+
requires_safety_checker=requires_safety_checker,
|
| 287 |
)
|
| 288 |
|
| 289 |
self.register_attn_hooks(unet)
|