Merge branch 'main' into JPPhoto-symmetry-enhancements

2026-01-22 16:08:02 -05:00 · 2023-03-10 12:35:15 -06:00
parent 9f8f85e922 685df33584
commit 9eba24f7ab
1 changed files with 1 additions and 1 deletions
--- 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: