View Source Bumblebee.Diffusion.StableDiffusion.SafetyChecker (Bumblebee v0.5.3)

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

References