Skip to content

GitLab

"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "fe234fc8cfae5fdbc9168994a0547c822631c3aa"
Unverified Commit f53352f7 authored by Clint Adams's avatar Clint Adams Committed by GitHub
Browse files

Set main_input_name in StableDiffusionSafetyChecker to "clip_input" (#7500) 



FlaxStableDiffusionSafetyChecker sets main_input_name to "clip_input".
This makes StableDiffusionSafetyChecker consistent.
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
parent 83ae24ce
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/pipelines/stable_diffusion/safety_checker.py
View file @ f53352f7
...@@ -31,6 +31,7 @@ def cosine_distance(image_embeds, text_embeds): ...@@ -31,6 +31,7 @@ def cosine_distance(image_embeds, text_embeds):
class StableDiffusionSafetyChecker(PreTrainedModel): class StableDiffusionSafetyChecker(PreTrainedModel):
config_class = CLIPConfig config_class = CLIPConfig
main_input_name = "clip_input"
_no_split_modules = ["CLIPEncoderLayer"] _no_split_modules = ["CLIPEncoderLayer"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment