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