Added image_encoder to init
Browse fileshttps://github.com/huggingface/diffusers/pull/5993#issuecomment-1833261167
- pipeline.py +4 -2
pipeline.py
CHANGED
@@ -273,6 +273,7 @@ class LLMGroundedDiffusionPipeline(StableDiffusionPipeline):
|
|
273 |
scheduler: KarrasDiffusionSchedulers,
|
274 |
safety_checker: StableDiffusionSafetyChecker,
|
275 |
feature_extractor: CLIPImageProcessor,
|
|
|
276 |
requires_safety_checker: bool = True,
|
277 |
):
|
278 |
super().__init__(
|
@@ -281,8 +282,9 @@ class LLMGroundedDiffusionPipeline(StableDiffusionPipeline):
|
|
281 |
tokenizer,
|
282 |
unet,
|
283 |
scheduler,
|
284 |
-
safety_checker,
|
285 |
-
feature_extractor,
|
|
|
286 |
requires_safety_checker=requires_safety_checker,
|
287 |
)
|
288 |
|
|
|
273 |
scheduler: KarrasDiffusionSchedulers,
|
274 |
safety_checker: StableDiffusionSafetyChecker,
|
275 |
feature_extractor: CLIPImageProcessor,
|
276 |
+
image_encoder: CLIPVisionModelWithProjection = None,
|
277 |
requires_safety_checker: bool = True,
|
278 |
):
|
279 |
super().__init__(
|
|
|
282 |
tokenizer,
|
283 |
unet,
|
284 |
scheduler,
|
285 |
+
safety_checker=safety_checker,
|
286 |
+
feature_extractor=feature_extractor,
|
287 |
+
image_encoder=image_encoder,
|
288 |
requires_safety_checker=requires_safety_checker,
|
289 |
)
|
290 |
|