From 5372800e60e3a68501c9c9fc24d3551124491b01 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Thu, 9 Mar 2023 17:38:58 -0500 Subject: [PATCH] Disable built-in NSFW checker on models converted with --ckpt_convert When a legacy ckpt model was converted into diffusers in RAM, the built-in NSFW checker was not being disabled, in contrast to models converted and saved to disk. Because InvokeAI does its NSFW checking as a separate post-processing step (in order to generate blurred images rather than black ones), this defeated the --nsfw and --no-nsfw switches. This closes #2836 and #2580. --- ldm/invoke/ckpt_to_diffuser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ldm/invoke/ckpt_to_diffuser.py b/ldm/invoke/ckpt_to_diffuser.py index 82ba73b0a4..a6bafc8c76 100644 --- a/ldm/invoke/ckpt_to_diffuser.py +++ b/ldm/invoke/ckpt_to_diffuser.py @@ -1002,7 +1002,7 @@ def load_pipeline_from_original_stable_diffusion_ckpt( tokenizer=tokenizer, unet=unet.to(precision), scheduler=scheduler, - safety_checker=safety_checker.to(precision), + safety_checker=None if return_generator_pipeline else safety_checker.to(precision), feature_extractor=feature_extractor, ) else: