AlanB commited on
Commit
a4d2d0a
·
1 Parent(s): d33f2b7

make requires_safety_checker a kwarg instead of a positional argument as it's more future-proof

Browse files
Files changed (1) hide show
  1. 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, text_encoder, tokenizer, unet, scheduler, safety_checker, feature_extractor, requires_safety_checker
 
 
 
 
 
 
 
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)