InvokeAI/invokeai/backend/model_management
Lincoln Stein 685df33584
fix bug that caused black images when converting ckpts to diffusers in RAM (#2914)
Cause of the problem was inadvertent activation of the safety checker.

When conversion occurs on disk, the safety checker is disabled during loading.
However, when converting in RAM, the safety checker was not removed, resulting
in it activating even when user specified --no-nsfw_checker.

This PR fixes the problem by detecting when the caller has requested the InvokeAi
StableDiffusionGeneratorPipeline class to be returned and setting safety checker
to None. Do not do this with diffusers models destined for disk because then they
will be incompatible with the merge script!!

Closes #2836
2023-03-10 18:11:32 +00:00
..
__init__.py fix call signature of import_diffuser_model() 2023-03-05 23:37:59 -05:00
convert_ckpt_to_diffusers.py fix bug that caused black images when converting ckpts to diffusers in RAM (#2914) 2023-03-10 18:11:32 +00:00
model_manager.py fix dangling reference to _model_to_cpu and missing variable model_description 2023-03-09 21:41:45 -05:00