src/diffusers/pipelines/stable_diffusion/safety_checker.py · 04f4bd54ea3126185ced2ffdf26f608dcd1db30e · renzhc / diffusers_dcu

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

Clint Adams authored Apr 29, 2024



FlaxStableDiffusionSafetyChecker sets main_input_name to "clip_input".
This makes StableDiffusionSafetyChecker consistent.
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Co-authored-by: YiYi Xu <yixu310@gmail.com>

f53352f7

safety_checker.py 5.63 KB

Replace safety_checker.py