AlanB commited on
Commit
e6d855a
1 Parent(s): a4d2d0a

Added image_encoder to init

Browse files

https://github.com/huggingface/diffusers/pull/5993#issuecomment-1833261167

Files changed (1) hide show
  1. 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