Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix "config attributes…not expected" diffusers warnings.

Browse Source
This commit is contained in:
Kevin Turner 2022-11-29 19:07:13 -08:00
parent 8157bff4bc
commit 7c558d5056
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ldm/invoke/generator/diffusers_pipeline.py
View File

@ -83,8 +83,10 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline):
scheduler: Union[DDIMScheduler, PNDMScheduler, LMSDiscreteScheduler],
safety_checker: Optional[StableDiffusionSafetyChecker],
feature_extractor: Optional[CLIPFeatureExtractor],
requires_safety_checker: bool = False
):
super().__init__(vae, text_encoder, tokenizer, unet, scheduler, safety_checker, feature_extractor)
super().__init__(vae, text_encoder, tokenizer, unet, scheduler,
safety_checker, feature_extractor, requires_safety_checker)
self.register_modules(
vae=vae,