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