View Source Bumblebee.Diffusion.StableDiffusion.SafetyChecker (Bumblebee v0.4.2)
A CLIP-based model for detecting unsafe image content.
This model is designed primarily to check images generated using Stable Diffusion.
Architectures
:base- the base safety detection model
Inputs
"pixel_values"-{batch_size, image_size, image_size, num_channels}Featurized image pixel values.
Configuration
:clip_spec- the specification of the CLIP model. SeeBumblebee.Multimodal.Clipfor details