diff --git a/invokeai/backend/model_management/convert_ckpt_to_diffusers.py b/invokeai/backend/model_management/convert_ckpt_to_diffusers.py index ad2c6afe9e..ae5550880a 100644 --- a/invokeai/backend/model_management/convert_ckpt_to_diffusers.py +++ b/invokeai/backend/model_management/convert_ckpt_to_diffusers.py @@ -1274,7 +1274,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: