"fair_dev/testing/testing.py" did not exist on "7abaa2bedb06a7a128ea805699fb37abdc1560ca"
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
......@@ -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"]
......
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