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: