From cc3d6ba91269e92ac4ec4faebc83a22dfbf77f7c Mon Sep 17 00:00:00 2001 From: Clint Adams Date: Wed, 27 Mar 2024 17:28:24 -0400 Subject: [PATCH] Set main_input_name in StableDiffusionSafetyChecker to "clip_input" FlaxStableDiffusionSafetyChecker sets main_input_name to "clip_input". This makes StableDiffusionSafetyChecker consistent. --- src/diffusers/pipelines/stable_diffusion/safety_checker.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/pipelines/stable_diffusion/safety_checker.py b/src/diffusers/pipelines/stable_diffusion/safety_checker.py index 6cc4d26f29b4..3e6dec3e0bff 100644 --- a/src/diffusers/pipelines/stable_diffusion/safety_checker.py +++ b/src/diffusers/pipelines/stable_diffusion/safety_checker.py @@ -31,6 +31,7 @@ def cosine_distance(image_embeds, text_embeds): class StableDiffusionSafetyChecker(PreTrainedModel): config_class = CLIPConfig + main_input_name = "clip_input" _no_split_modules = ["CLIPEncoderLayer"]